博主在配置pytorch之前已经安装好了anaconda,没有安装过的小可爱要先安装一下anaconda哦~
话不多说,进入正题
Pytorch环境配置:
1.新建虚拟环境:
conda create -n pytorch python=3.6(pytorch中指定你虚拟环境的名字,python后面的数字是虚拟环境中python的版本)
2.进入新建的虚拟环境中 conda activate pytorch
3.配置清华镜像,因为在pytorch官网进行下载安装的话网速会非常慢,不一定能装好。
命令行分别输入:
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
4.之后进入到pytorch官网,查看与自己Linux服务器、CUDA版本相对应的安装指令
5.在命令行中输入相应的安装指令:
conda install pytorch torchvision cudatoolkit=10.1
!!!注意把指令后的 –c pytorch去掉,因为加上的话就会指定到pytorch官网下载安装包,而我们想要从清华镜像中下载相应的安装包
6.一路yes,下