Sop:Ubuntu中安装pytorch cuda cudnn

问题

• 在Ubuntu中安装pytorch系列

解决

step1 双系统安装

• 本文适合UEFI和单磁盘

• 查询win+r msinfo32
  

• 查询 cmd 中nvidia-smi cuda最高版本（安装时一般版本稍低一点）
  

• 移步软件管家教程
  注意不要跳步

• 换下载路径Software & Updates中

• 
  

step2 anaconda安装

• 清华源链接

• 直接在虚拟机浏览器中下载

• 
  

• 找最新的版本即可
  

• 在安装包路径下打开终端并输入

bash xxx.sh

conda create -n aalh python=3.8  //创建虚拟环境
conda activate aalh              //切换

• 换下载路径

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
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/peterjc123/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

conda install pytorch torchvision torchaudio cudatoolkit=11.1  //小于11.2 查询值

step3 装驱动

sudo apt purge nvidia-*  #删除旧的NVIDIA驱动
sudo add-apt-repository ppa:graphics-drivers/ppa
sudo apt update && sudo apt upgrade//更新
ubuntu-drivers devices  #查看自己的显卡及可以安装的驱动版本
sudo apt install nvidia-driver-410  #根据自己的驱动建议安装
sudo reboot

step4 验证

conda activate aalh              //切换
python         
import torch 
torch.cuda.is_available()

• 最后True表示ok