Ubuntu18.04安装anaconda+CUDA+cudnn+pytorch过程记录

Ubuntu18.04安装anaconda+CUDA+pytorch过程记录

  • anaconda
  • CUDA
  • cudnn
  • pytorch

anaconda

问题:安装完anaconda后,在terminal中输入anconda,查无此命令。
原因:没有配置环境变量。
解决:在桌面进入terminal,输入vim .bashrc,编辑此文件。进入文件后,按i进入插入模式,将光标移到文件末尾,另起一行,输入export PATH=~/anaconda3/bin:$PATH,按esc退出插入模式,输入:wq!,保存并退出文件。
注:vim的相关操作,参考https://blog.csdn.net/evergreenswj/article/details/86573371。

CUDA

问题:CUDA10.2下载成功,安装出现问题。
原因:CUDA10版本集成了nvidia驱动程序,与系统本身就有的驱动程序冲突,所以安装的时候才会出现,让你先卸掉驱动。
解决:安装10.1版本,这一版本安装的时候可以选择不安装驱动程序。如图,将光标移到前两行,按空格取消选择,再选择安装。
Ubuntu18.04安装anaconda+CUDA+cudnn+pytorch过程记录_第1张图片
安装成功结果图如下:
Ubuntu18.04安装anaconda+CUDA+cudnn+pytorch过程记录_第2张图片
问题:配置CUDA环境变量。
解决:与上面配置anaconda环境变量相同,也是在".bashrc"文件末尾另起一行,添加如下代码:

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

退回到terminal输入source .bashrc执行文件,使其生效。最后输入nvcc --version测试cuda是否安装成功,成功结果图如下:
在这里插入图片描述

cudnn

在官网选择和cuda对应版本的cudnn,下载.deb文件,执行sudo dpkg -i libcudnn7_7.4.1.5-1+cuda10.1_amd64.deb命令,安装完成。
如何确认cudnn安装成功?

pytorch

问题:创建环境的时候下载python慢;下载pytorch慢。
解决:配置清华镜像源:

conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

问题:下载到torchvision卡住。
原因:从官网copy的命令末尾-c pytorch中的"-c"参数指定channel下载,用不上配置的清华镜像源。
解决:去掉-c pytorch

你可能感兴趣的:(Ubuntu18.04安装anaconda+CUDA+cudnn+pytorch过程记录)