ubantu系统下安装pytorch

配置环境的步骤&＃xff1a;
1、下载anaconda

wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/Anaconda3-5.3.1-Linux-x86_64.sh


2、解压安装anaconda&＃xff0c;一直点yes

bash Anaconda3-5.3.1-Linux-x86_64.sh


3、激活安装好的配置环境

source ~/.bashrc


4、检查是否安装成功

conda


出现以下情况说明安装成功

若是提示没有conda命令

则在自己的.bashrc文件&＃xff08;一般在/home/user下&＃xff09;后面加上export PATH&＃61;/home/sun/anaconda3/bin:$PATH 并进行保存 #sun根据自己的用户名选择性进行修改
再执行

source ~/.bashrc


5、创建虚拟环境

conda create --name pytorch python&＃61;3.6

6、激活虚拟环境

source activate pytorch


一些虚拟环境的基本操作&＃xff1a;

source activate tf #激活tf环境
source deactivate tf #退出tf环境
conda remove --name tf --all #删除tf环境&＃xff08;全部删除&＃xff09;
conda env list#查看有哪些虚拟环境
conda list #查看环境下有哪些包


7、在虚拟环境中安装Pytorch&＃xff0c;直接安装速度太慢&＃xff0c;先配置的清华的通道

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/


配置查看&＃xff1a;

conda config --show

显示如下说明配置成功

8、安装Pytorch
先查看驱动版本&＃xff1a;

nvidia-smi


在这里插入图片描述
可以看到cuda版本为10.2

或者可以根据驱动版本进行查看
在这里插入图片描述

conda install pytorch torchvision cudatoolkit&＃61;10.2


9、最后查看GPU是否可用&＃xff0c;输出为true安装成功

python
import torch
torch.cuda.is_available()


在这里出现了一个问题&＃xff0c;就是torch.cuda.is_available()显示False&＃xff0c;上网查了原因后&＃xff0c;才知道是pytorch版本和cuda版本不一致&＃xff0c;这里cuda版本是10.2&＃xff0c;但是我默认安装的pytorch版本是0.1的&＃xff0c;完全不够&＃xff0c;查阅之后&＃xff0c;知道相应的pytorch版本是1.6.0的。需以下步骤处理
卸载pytorch

conda uninstall pytorch


指定pytorch版本进行重安装

conda install pytorch&＃61;1.6.0 torchvision cudatoolkit&＃61;10.2


最后结果显示这样代表安装成功
另外注意cuda&＃xff0c;torch&＃xff0c;还有torchvision三者的关系为
具体见网址https://pytorch.org/get-started/previous-versions/
或者https://download.pytorch.org/whl/torch_stable.html