首先根据自己需要安装对应版本的anaconda 。
官网地址:https://www.anaconda.com/products/individual(下载速度较慢)
清华源:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/(下载快速且不易出错)
下载过程中如果没有框选自动配置变量的,下载完成后,需要配置环境变量。
我的安装在D盘Anaconda目录下,我的环境变量配置如下:
第一个变量将你的anaconda的安装地址填入;
第二个变量是anaconda下scripts文件夹目录;
第三个变量是anaconda下bin文件夹目录。
配置好环境后就可以在anaconda环境下安装pytorch啦!
由于环境中需要安装很多软件,他们的Python的关联模块或版本会有所差异。如果全部安装到一个环境中会引起冲突。所以选择虚拟环境能很好地避免环境之间的冲突。
1、打开Anaconda Prompt
2、建立并切换到指定工作目录
md \pythonwork
cd \pythonwork
3、建立Pytorch Anaconda虚拟环境
conda create --name pytorch python=3.7 anaconda
这里我选择的是Python3.7,大家可以自行选择
4、启动Pytorch Anaconda虚拟环境
activate pytorch
5、安装Pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
【注意】关于CUDA版本选择
以上步骤第五条我选择的是CUDA10.1。确保电脑有合适的GPU
(1)搜索并打开NVIDA控制面板
(2)选择帮助-系统信息
(3)查看驱动程序版本
(4)记下来查阅对应版本
https://docs.nvidia.com/cuda/cuda-toolkit-release-notes/index.html
6、测试
命令行输入python进入Python,输入以下代码:
import torch
import torchvision
print(torch.__version__)
成功打印,测试成功,使用exit()退出Python
7、关闭Pytorch Anaconda虚拟环境
conda deactivate
注意
如果下载速度过慢要注意自己的下载源是否更改,可以更换清华源,或者大家懒得换源的话,可以直接翻墙食用。