pytorch搭建

小插曲：以用户登陆xshell后，只显示$。切换到root用户下执行#usermod -s /bin/bash username

一、安装anaconda3（如果需要用python2，用conda新建python2的虚拟环境即可，conda create -n py2 python=2.7）

用命令安装:bash Anaconda3-5.2.0-Linux-x86_64.sh 

安装之后 source ~/.bashrc让用户环境变量生效，因为安装anaconda的过程中选择了自动写到环境变量PATH中

conda create -n pytorch python=3.6

source activate pytorch(退出是deactivate)

添加国内源：

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

二、安装pytorch

接着新建虚拟环境

https://pytorch.org/网站找出所用cuda版本，系统型号，会在下面显示命令，直接输入即可

conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

三、测试pytorch是否安装成功

import torch

print(torch.cuda.is_available())如果输出我True，说明安装成功