Anaconda + Pytorch 安装踩坑

在对硬盘空间的最后压榨和对校网的无尽折磨后,终于装上了..

测试系统信息

  • Windows 10 家庭中文版 21H1
  • NVIDIA GeForce RTX 2060 with Max-Q Design
  • AMD Ryzen 7 4800HS with Radeon Graphics 2.90 GHz
  • NVCUDA64.DLL 11.6.127 (可在 NVIDIA控制面板 – 帮助 – 系统信息 – 组件 里查询)

安装ANACONDA

Anaconda是一个开源的Python发行版本,内含conda, python, numpy等180多个科学包及其依赖项。

官网下载安装包安装即可。安装成功后可以使用Anaconda Navigator通过图形界面创建虚拟python环境,或使用Anaconda Prompt以命令行的方式创建虚拟python环境。

如果在安装时勾选了将其添加至环境变量,也可以通过cmd或powershell方便地调用python,conda等指令。不过,考虑到我已经装了python的另一个版本,为了避免重复安装其他软件(例如VSCode的一些插件)的依赖包,还是选择通过Anaconda Prompt调用吧。

ANACONDA换源

这里使用anaconda | 镜像站使用帮助 | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

按照提示修改.condarc文件,并执行以下命令即可

conda clean -i

安装CUDA*

(Anaconda自带了pytorch与cudatoolkit,但不知为啥,实测无法成功调用cuda…还是老老实实地把cuda下下来吧…

注意到我的cuda版本是11.6.x,因此下载相应版本的CUDA Toolkit 11.6 Update 1 Downloads | NVIDIA Developer

(网上的教程大都给了这步,后来发现这一步骤似乎不是必要的…根据以下博客,貌似只要通过conda安装相应版本的cudatoolkit即可…

一文讲清楚CUDA、CUDA toolkit、CUDNN、NVCC关系健0000的博客-CSDN博客cudatoolkit和cuda有关系吗

安装pytorch

首先创建一个虚拟python环境,这里命名为test。

conda create -n test

激活test环境。

conda activate test

根据官网给出的提示,安装pytorch

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

这一步大约需要下载几个G的文件。由于服务器在国外,需要等待一段时间。期间如果出现网络报错,可以尝试重复该命令。

测试pytorch

打开python

python

输入以下内容

import torch
x = torch.rand(5, 3)
print(x)

若会出现以下类似结果

tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

说明pytorch已成功安装。

输入以下内容

import torch
torch.cuda.is_available()

如果返回True,那么Congratulation!cuda已正确配置。

发表评论

%d 博主赞过: