ubuntu16.04 卸载安装cudn cudnn pytorch

目录

一、base环境下安装cudn和cudnn

1.卸载旧cuda9.0（不需要则跳过）

2.安装新cuda10.1

3.删除旧版本的cudnn

二、在pytorch虚拟环境中安装cudn和cudnn 

三、运行出错记录

四、参考链接

---

一、base环境下安装cudn和cudnn

1.卸载旧cuda9.0（不需要则跳过）

sudo /usr/local/cuda-9.0/bin/uninstall_cuda_9.0.pl

卸载之后，会发现/usr/local/cuda-9.0目录下仍然有文件存在，仍然删除即可

sudo rm -rf /usr/local/cuda-9.0

2.安装新cuda10.1

找自己对应版本下载 https://developer.nvidia.com/cuda-toolkit-archive

2.1 安装cuda

下载安装包后执行

sudo sh cuda_10.1.105_418.39_linux.run

2.2 修改环境变量

打开.bashrc文件 

 vi ～/.bashrc

(或者直接找，在/home/ly(自己对应文件)，如果没有找到，则按Ctrl+H键显示隐藏文件)

export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda-10.1/lib64
export PATH=$PATH:/usr/local/cuda-10.1/bin
export CUDA_HOME=$CUDA_HOME:/usr/local/cuda-10.1

之后保存，执行环境环境变量

source ~/.bashrc 

查看cuda版本

cat /usr/local/cuda/version.txt

3.删除旧版本的cudnn

sudo rm -rf /usr/local/cuda/include/cudnn.h
sudo rm -rf /usr/local/cuda/lib64/libcudnn

4.安装新版本的cudnn

下载 https://developer.nvidia.com/rdp/cudnn-archive

解压

tar -xvf cudnn-8.0-linux-x64-v5.1.tgz

安装 

sudo cp cuda/include/cudnn.h /usr/local/cuda/include/
sudo cp cuda/lib64/libcudnn* /usr/local/cuda/lib64/
sudo chmod a+r /usr/local/cuda/include/cudnn.h
sudo chmod a+r /usr/local/cuda/lib64/libcudnn*

查看当前cudnn的版本

cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2

ok

二、在pytorch虚拟环境中安装cudn和cudnn 

1.首先创建pytorch虚拟环境

conda create -n pytorch python=3.7

2.进入pytorch虚拟环境

activate pytorch

3.安装cuda

conda install cudatoolkit=10.1 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/linux-64/

4.安装cudnn

conda install cudnn=7.0.5 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/

5.安装pytorch

conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=10.1 -c pytorch

或者找自己的版本 https://pytorch.org/get-started/previous-versions/

三、运行出错记录

1. No module named 'tensorboardX'
   pip install tensorboardX
2. No module named 'cv2'
   pip install opencv-python
3. No module named 'pycocotools'
   pip install pycocotools  
4. No module named 'terminaltables'
   pip install terminaltables

四、参考链接

https://blog.csdn.net/YPP0229/article/details/108939159

https://blog.csdn.net/benbenls/article/details/103516708

https://blog.csdn.net/xue_csdn/article/details/100748309

https://blog.csdn.net/kaixinjiuxing666/article/details/80321124