Ubuntu18.04 cuda10.1安装与卸载,NVIDIA驱动安装
1 cuda 10.0卸载
sudo /usr/local/cuda-10.0/bin/uninstall_cuda_10.0.pl
删除残余
sudo /usr/local/cuda
卸载完成!
2 安装nvidia驱动
sudo ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
但是安装完后重启系统进不去
这里可以进入advanced options模式
选择recovery模式,选择dpkg进行自动修复,需要联网重启,即可
3安装cuda 10.1
chmod +x cuda_10.1.168_418.67_linux.run
sudo ./cuda_10.1.168_418.67_linux.run
驱动不选择
添加环境变量
sudo gedit ~/.bashrc
export CUDA_HOME=/usr/local/cuda
export PATH=$PATH:$CUDA_HOME/bin
export LD_LIBRARY_PATH=/usr/local/cuda-10.1/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
保存退出
source ~/.bashrc
验证是否成功
cd /usr/local/cuda/samples/1_Utilities/deviceQuery
sudo make
./deviceQuery
接下来解压cudnn
用下列命令复制cudnn目录
sudo cp cudnn-10.1-linux-x64-v7.6.5.32/cuda/lib64/* /usr/local/cuda-10.1/lib64/
sudo cp cudnn-10.1-linux-x64-v7.6.5.32/cuda/include/* /usr/local/cuda-10.1/include/
查看版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
4 Anconda 创建环境出错
改成 python==3.8就可以了
5 安装torch
激活环境后
pip install torch-1.8.0+cu101-cp38-cp38-linux_x86_64.whl