服务器ubuntu20.04配置pytorch环境

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

安装Anaconda3

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

安装Nvidia驱动

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

sudo ubuntu-drivers devices
sudo ubuntu-drivers autoinstall

安装成功后重启

sudo reboot

输入

nvidia-smi

就可以查看啦

CUDA11.4降11.0

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

Nvidia官网给出的CUDA11.0 runfile安装方式：