基于cuda12.1安装pytorch-gpu

最近翻新了显卡,又得重新装一次环境。因课题需要,特别需要pytorch和tensorflow的gpu版本,为了方便大家装环境,故将这次重装pytorch-gpu过程给大家做个参考,期间需要的安装包也放在某盘了,需要可以自取。

  1. 安装Anaconda(如果未安装的话)。Anaconda是一个Python的科学计算发行版,包含了大量常用的数据科学和机器学习库。前往Anaconda网站的下载页面(https://www.anaconda.com/products/individual#Downloads)下载对应的安装程序,并按照安装指南进行安装。

  2. 安装合适的CUDA驱动程序和cuDNN库。首先确定你的显卡型号,然后从NVIDIA官方网站下载并安装对应的CUDA驱动程序。接着从NVIDIA开发者网站下载并安装对应版本的cuDNN库。
    安装更低版本的CUDA驱动程序和对应cuDNN库。安装CUDA过程一直点下一步即可。安装好CUDA后,将cuDNN压缩包解压至CUDA路径。
    命令行nvidia-smi

     

     要下载比对应版本更低的版本,我这是12.2,所以下载了12.1,下载链接在末尾。

  3. 打开Anaconda Prompt(或者其他任何终端),创建一个新的虚拟环境。可以使用以下命令在Anaconda Prompt中创建一个名为pytorch的虚拟环境:

    conda create --name pytorc

你可能感兴趣的:(pytorch,人工智能,python)