Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装

Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装

文章目录

  • Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装
  • 一、CUDA10.0安装
  • 二、cuDNN v7.6.5安装
    • 下载链接:
  • 二、Anaconda3-5.3.0安装
    • 下载链接:
  • 四、pycharm安装
    • 下载链接:
  • 五、pythorch安装
  • 总结


一、CUDA10.0安装

https://zhuanlan.zhihu.com/p/47330858 --按照这位博主写的,我补充了一下

这CUDA的下载地址https://developer.nvidia.com/cuda-toolkit-archive,在里面选择适合你系统的版本(参考上一篇),我选择CUDA10.0,如下图。
Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装_第1张图片点击后进入下图页面,installer Type 选择deb(local)(官网建议),其它根据自己的版本来选择。
Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装_第2张图片代码如下:

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

二、cuDNN v7.6.5安装

下载链接:

https://developer.nvidia.com/cudnn,选择cuDNN Library for Linux下载即可。
Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装_第3张图片
代码如下:

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

Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装_第4张图片

二、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! 即安装成功。

Ubuntu+CUDA+cuDNN+Anaconda+pycharm+pytorch安装_第5张图片
重启终端,即可使用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安装

参考以下博主的安装过程:
https://blog.csdn.net/m0_37758063/article/details/110734347?ops_request_misc=%257B%2522request%255Fid%2522%253A%2522161503794216780357240994%2522%252C%2522scm%2522%253A%252220140713.130102334.pc%255Fall.%2522%257Drequest_id=161503794216780357240994biz_id=0utm_medium=distribute.pc_search_result.nonetaskblog2allfirst_rank_v2~rank_v29-2-110734347.pc_search_result_hbase_insertutm_term=pythorch%E5%AE%89%E8%A3%85%E6%95%99%E7%A8%8BUBUNTU


总结

要显示显卡信息,在终端输入:

sudo lshw -c video

要显示CUDA信息,在终端输入:

nvcc -V

查看Phthon版本,在终端输入:

python --version

验证pytorch是否安装成功,在终端输入:

python

此时进入python环境。然后,继续输入

import torch

import torchvision

不报错就表明安装成功。

你可能感兴趣的:(python,linux,ubuntu,cuda)