1、显卡
nvidia470显卡
chmod +x NVIDIA-Linux-x86_64-470.86.run
sudo ./NVIDIA-Linux-x86_64-470.86.run --no-opengl-files --no-x-check --no-nouveau-check
2、环境要求
cuda 11.1
cudnn 8.05
torch==1.7.1+cu110
torchvision==0.8.2+cu110
torchaudio==0.7.2
安装pytorch工具包时 在命令后面加入以下命令
-f https://download.pytorch.org/whl/torch_stable.html
安装其他工具包时添加镜像源
-i https://pypi.tuna.tsinghua.edu.cn/simple
3、添加pytorch清华镜像源
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
然后conda install xxx (去掉后面的 -c pytorch)