服务器ubuntu20.04配置pytorch环境

• 安装Anaconda3
• 安装Nvidia驱动
• CUDA11.4降11.0
• 安装pytorch1.7.0

因为是远程配置服务器，所以全程是终端命令配置

wget下载

wget -P /tmp https://repo.anaconda.com/archive/Anaconda3-2020.07-Linux-x86_64.sh

安装

bash /tmp/Anaconda3-2020.07-Linux-x86_64.sh

按enter浏览完许可证
输入yes接受
enter确认安装
安装完成输入yes运行conda init

安装完毕终端输入

source ~/.bashrc

终端输入进行测试

conda list

懒人方法
自动找合适的驱动进行安装

sudo ubuntu-drivers devices
sudo ubuntu-drivers autoinstall

安装成功后重启

sudo reboot

输入

nvidia-smi

就可以查看啦

目前pytorch还不支持CUDA11.4，所以降到11.0，其他版本也可以

Nvidia官网给出的CUDA11.0 runfile安装方式：
https://developer.nvidia.com/cuda-11.0-download-archive?target_os=Linux&target_arch=x86_64&target_distro=Ubuntu&target_version=2004&target_type=runfilelocal

wget http://developer.download.nvidia.com/compute/cuda/11.0.2/local_installers/cuda_11.0.2_450.51.05_linux.run
sudo sh cuda_11.0.2_450.51.05_linux.run

进入提示后选continue
许可证界面输入accept
进入安装选项界面：去掉driver选项（[ ]内无X表示已去掉）然后选择install安装

安装结束后配置环境变量
终端输入
vim ~/.bashrc

在文件结尾插入（这里如果装了别的版本CUDA，把11.0改成相应版本即可）

export PATH=$PATH:/usr/local/cuda-11.0/bin
export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-11.0/lib64
export LIBRARY_PATH=$LIBRARY_PATH:/usr/local/cuda-11.0/lib64

esc 后输入：wq保存退出

进入终端输入

source ~/.bashrc

配置完成
可输入
nvcc -V
测试是否配置成功
如果显示的不是安装的CUDA版本
可终端输入重启

sudo reboot

重启后再次测试

根据pytorch官方给出的方法自行选择版本进行安装 https://pytorch.org/get-started/previous-versions/

终端输入（这里我安装的1.7.0，CUDA11.0的）

conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

输入y进行安装
安装成功后
终端输入

python3

进入python模式测试
输入

import torch
torch.__version__

查看pytorch版本