Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装
文章目录
一、CUDA10.0安装
https://zhuanlan.zhihu.com/p/47330858 --按照这位博主写的,我补充了一下
这CUDA的下载地址https://developer.nvidia.com/cuda-toolkit-archive,在里面选择适合你系统的版本(参考上一篇),我选择CUDA10.0,如下图。
点击后进入下图页面,installer Type 选择deb(local)(官网建议),其它根据自己的版本来选择。
代码如下:
sudo dpkg -i cuda-repo-ubuntu1604-10-0-local-10.0.130-410.48_1.0-1_amd64.deb
sudo apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
sudo apt-get update
sudo apt-get install cuda
安装完后通过下面的命令查看安装情况。
nvcc -V
二、cuDNN v7.6.5安装
下载链接:
https://developer.nvidia.com/rdp/cudnn-archive,选择cuDNN Library for Linux下载即可。
代码如下:
tar -xzvf cudnn-10.0-linux-x64-v7.6.5.32.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 /usr/local/cuda/lib64/libcudnn*
完成后,通过下面的命令查看安装情况,如果结果如下图逐行显示版本号,则安装成功。
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
二、Anaconda3-5.3.0安装
Anaconda与Python安装版本对应关系:见官网
下载链接:
https://www.anaconda.com/distribution/#download-section 下载Anaconda3-5.3.0,对应python3.7版本
终端输入代码:
cd Downloads #文件路径
bash Anaconda3-5.3.0-linux-x86_64.sh
注意:提示信息“Do you wish to proceed with the installation of Microsoft VSCode? [yes|no]”,输入no;
出现Thank you for installing Anaconda3! 即安装成功。
重启终端,即可使用Anaconda3;
四、pycharm安装
pycharm是目前使用最多的python IDE,功能强大,界面简洁美观,安装简单。
下载链接:
http://www.jetbrains.com/pycharm/download/#section=linux
下载Community版 for Linux。并将下载好的文件copy到想安装的目录下。
执行以下代码:
cd Downloads
tar xfz pycharm-community-2020.3.3.tar.gz
cd pycharm-community-2020.3.3/bin
./pycharm.sh
五、pythorch安装
总结
要显示显卡信息,在终端输入:
sudo lshw -c video
要显示CUDA信息,在终端输入:
nvcc -V
查看Phthon版本,在终端输入:
python --version
验证pytorch是否安装成功,在终端输入:
python
此时进入python环境。然后,继续输入
import torch
import torchvision
不报错就表明安装成功。