- anaconda下Pytorch的安装
- 检查是否已经安装CUDA
- 安装Pytorch
- 查看安装命令
- 创建新环境
- 安装pytorch
安装gpu pytorch是不需要自己手动安装cuda和cudnn,小编当时找了很多教程都是先安装CUDA再安装CUDNN最后安装pytorch,整了好久也没看明白,网友评论真是给我当头一棒。果真,大部分电脑应该已经有CUDA了,所以不用提前安装,大家一定要先看看自己有没有在安装哦。
win+R打开cmd界面
输入:
nvidia-smi
可查看是否安装CUDA以及其版本。
CUDA的详细介绍可见:CUDA——“从入门到放弃”
由图可见,小编的电脑里已经有CUDA了,并且版本为10.2。fine,开启下一步!
打开官网PyTorch
->Install
根据自己电脑的配置以及需求进行选择(CUDA的版本在步骤一里头已经查看过了):
官网非常好心的已经给我们生成了安装指令即Run this Command中所给出的指令。小编的安装命令为:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch创建新环境
打开cmd,在命令行输入:
create conda -n pytorch1.7 python=3.7
激活虚拟环境pytorch1.7。输入命令:
conda activate pytorch1.7安装pytorch
由于pytorch非常的大,直接运行官网给的运行命令会龟速安装,因此,我们需要更改源来安装,小编选的是清华源,网址如下:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
在界面中找到自己对应的系统并点开,小编是win-64:
点开后复制镜像源的地址,小编的地址为:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
然后,我们打开pytorch的安装命令并将其末尾的pytorch删掉并跟改为镜像源的地址:
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
安装完毕!



