ubuntu20.04 安装Anaconda3+CUDA+cudnn+Pytorch

yishuihan-Oliver

已于 2022-08-23 09:58:25 修改

阅读量6.5k

点赞数 7

分类专栏：环境搭建机器学习_深度学习文章标签： pytorch 深度学习

于 2021-12-28 23:09:13 首次发布

本文链接：https://blog.csdn.net/yishuihanq/article/details/122204902

版权

机器学习_深度学习同时被 2 个专栏收录

26 篇文章 11 订阅

订阅专栏

环境搭建

7 篇文章 1 订阅

订阅专栏

ubuntu20.04 安装Anaconda3+CUDA+cudnn+Pytorch

Ubuntu GPU驱动 / CUDA版本 / CuDNN 版本，都要相互关联，版本不对应的话，就会出错。
版本确认顺序：CUDA版本--> CuDNN版本--> GPU版本 。
安装顺序：GPU版本-->CUDA版本-->CuDNN版本 。

1、安装Anaconda：

打开终端,转到自己的下载目录下（命令为：cd /home/oliver/下载
），然后输入以下命令：
bash Anaconda3-2020.11-Linux-x86_64.sh
终端中查看Anaconda版本：conda -V 或者 conda --version

官网：Anaconda | The World's Most Popular Data Science Platform

Old package lists — Anaconda documentation

历史版本地址：Index of /

2、安装NVIDIA驱动：(我选用的方法2)

NVIDIA驱动下载官网：官方 GeForce 驱动程序 | NVIDIA

旧版本的NVIDIA驱动：Linux AMD64 Display Driver Archive | NVIDIA

（方法1）查看可安装的驱动：ubuntu-drivers devices
自动安装推荐的驱动：sudo ubuntu-drivers autoinstall #自动安装推荐驱动，安装完成后会需要定一个密码，一会儿要用，可以设置和开机密码一样
重启电脑：sudo reboot

（方法2）也可以在系统的“软件和更新”下的“附加驱动”里直接选择合适的驱动进行“应用更改”，完成后重启电脑。

（方法3）在官网下载对应的驱动到本地，在本地安装。进入我们保存下载 NVIDIA 驱动的目录，默认是：Downloads/ 目录，cd Downloads/，然后执行：bash NVIDIA-Linux-x86_64-418.43.bin。注意：NVIDIA-Linux-x86_64-418.43.bin 要根据你下载的驱动文件的名字相应改动。安装过程中都点同意即可。

查看NVIDIA显卡驱动是否安装成功：nvidia-smi
查看NVIDIA显卡配置选项：nvidia-settings

3、安装CUDA：(我选用的方法2)

官网地址：https://developer.nvidia.com/cuda-toolkit-archive

CUDA最新版下载地址：CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developer

以前版本的CUDA：CUDA Toolkit Archive | NVIDIA Developer

(方法1)系统默认安装CUDA
sudo apt install nvidia-cuda-toolkit （默认安装）
(方法2)下载CUDA到本地安装
打开终端,转到自己的下载目录下，然后输入以下命令：(本地安装指引)
chmod +x cuda_11.1.0_455.23.05_linux.run（有的博客中没有这一句）
sudo sh cuda_11.1.0_455.23.05_linux.run

(方法3)本地安装（本人没试过）
Installation Instructions:(本地安装指引)
`sudo dpkg -i cuda-repo-ubuntu1810-10-1-local-10.1.105-418.39_1.0-1_amd64.deb`
`sudo apt-key add /var/cuda-repo-<version>/7fa2af80.pub`
`sudo apt-get update`
`sudo apt-get install cuda`

添加环境变量，打开profile文件：sudo gedit /etc/profile
在末尾处添加以下内容(注意不要有空格，不然会报错)：
export PATH=/usr/local/cuda-11.1/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda-11.1/lib64$LD_LIBRARY_PATH

安装完，重启之后，查看CUDA版本：nvcc -V 或者 nvcc --version

4、安装对应的cuDNN

官网：NVIDIA cuDNN | NVIDIA Developer

以前的版本：cuDNN Archive | NVIDIA Developer

下载cuDNN到本地安装
打开终端,转到自己的下载目录下，然后输入以下命令：(本地安装指引)
sudo dpkg -i libcudnn8_8.1.1.33-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8-dev_8.1.1.33-1+cuda11.2_amd64.deb
sudo dpkg -i libcudnn8-samples_8.1.1.33-1+cuda11.2_amd64.deb

5、安装Pytorch：(我选用的方法4，另外三种都没成功)

进入官网选择适合当前机器环境的Pytorch

官网地址：Start Locally | PyTorch

旧版本：Previous PyTorch Versions | PyTorch

清华源：Index of /anaconda/cloud/pytorch/linux-64/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

离线安装时，对应cuda版本的torch和torchvision下载地址：

（方法1）官网conda安装，打开终端，输入以下命令：
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge（这个官方太慢）
（方法2）也可以添加清华镜像源(https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/linux-64/)下载，打开终端，输入以下命令：
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes
配置完成后运行的command命令改为，这里删除了 -c pytorch
conda install pytorch torchvision cudatoolkit=10.2

conda install pytorch torchvision cudatoolkit=11.1
（方法3）官网pip安装：
pip install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

sudo pip3 install torch==1.8.1+cu111 torchvision==0.9.1+cu111 torchaudio==0.8.1 -f https://download.pytorch.org/whl/torch_stable.html

（方法4）本地pip安装pytorch
下载与cudatoolkit=11.1相对应的pytorch torchvision torchaudio 到本地安装，地址：https://download.pytorch.org/whl/torch_stable.html
下载相对应的以下三个文件：
torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl（https://download.pytorch.org/whl/cu111/torch-1.8.1%2Bcu111-cp38-cp38-linux_x86_64.whl）
torchvision-0.9.1+cu111-cp38-cp38-linux_x86_64.whl
torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl（https://download.pytorch.org/whl/torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl）
转到自己的下载目录下，然后输入以下命令：(本地安装指引)
pip install torch-1.8.1+cu111-cp38-cp38-linux_x86_64.whl
pip install torchvision-0.9.1+cu111-cp38-cp38-linux_x86_64.whl
pip install torchaudio-0.8.1-cp38-cp38-linux_x86_64.whl

6、测试pytorch是否安装成功

打开终端下依次输入以下命令：
python #显示python版本
import torch
import torchvision
torch.cuda.is_available()

/----------------------------------------/
在主目录下新建文件夹：cd ~
mkdir software

pip install 待安装包名.whl

卸载：
pip uninstall torchvision
pip uninstall torch
/----------------------------------------/

ubuntu20.04安装OpenCV-python

(我选用的方法1-Option 1)
（方法1）从官网安装
去官网：https://pypi.org/project/opencv-python/，查看安装方法：
Option 1 - Main modules package: pip install opencv-python
Option 2 - Full package (contains both main modules and contrib/extra modules): pip install opencv-contrib-python (check contrib/extra modules listing from OpenCV documentation)

(方法2)可从默认的Ubuntu 20.04存储库安装OpenCV。
要安装它，请运行：
sudo apt update
sudo apt install libopencv-dev python3-opencv
上面的命令将安装OpenCV所需的所有软件包。

通过导入cv2模块并打印OpenCV版本来验证安装：
python3 -c "import cv2; print(cv2.__version__)"

在撰写本文时，存储库中的版本为4.2：
4.2.0python

重要参考：【Ubuntu18.04】Ubuntu18.04 + RTX3070 安装 Anconda3+cuda11.1+cuDNNv8.1.0+Pytorch - 知乎

【Ubuntu18.04】Ubuntu18.04 + RTX3070 安装 Anconda3+cuda11.1+cuDNNv8.1.0+Pytorch

笙箫绵绵

机器人工程师

0. 背景阅读

一开始我只是想要安装 pytorch，并不了解 cuda, cudnn 和 anaconda 之间的关系。

cuda 和 cudnn 之间的关系：CUDA 与 cuDNN，cuda 是螺丝刀，cudnn 是扳手，这两个工具在 GPU 加速运算中都是需要的，安装 cuda 并不会附带 cudnn，两个需要分别安装。
anaconda 是一个 python 的发行版，包括了 python 和很多常见的软件库, 和一个包管理器 conda。常见的科学计算类的库都包含在里面了，使得安装比常规 python 安装要容易。主要是！装了 anaconda 就不需要单独装 python 了。因为==Anaconda 就是用来管理我们不同版本的 python 环境的。==[2]

1. 安装显卡驱动

方法参考我之前的这篇博客：ubuntu18.04 下安装英伟达显卡驱动
检查驱动是否安装成功： sudo nvidia-smi 显示如下则说明安装成功：

2. gcc 升级

Ubuntu18.04 默认是 7.5.0 版本的 gcc，查看 gcc 版本：

gcc --version

到英伟达官网看最新的 CUDA11 支持的 gcc 版本，2021 年 3 月 18 日支持到 9.x 版本的 gcc。

2.1 安装 gcc 和 g++

sudo add-apt-repository ppa:ubuntu-toolchain-r/test
sudo apt-get update
sudo apt-get install gcc-9
sudo apt-get install g++-9

2.2 进入/usr/bin/把 gcc/g++文件夹删除或备份，建立链接到 gcc-5.4/g+±5.4

cd /usr/bin
sudo rm gcc
sudo ln -s gcc-9 gcc
sudo rm g++
sudo ln -s g++-9 g++

2.3 查看 gcc 版本号

gcc -v

3. 安装 anaconda3

3.1 下载

Anaconda3 下载官网地址：https://www.anaconda.com/products/individual

3.2 安装

Anaconda3-2020.02-Linux-x86_64.sh

3.3 修改终端默认的 python 版本

sudo gedit ~/.rc

在.rc 文件末尾写上

export PATH="/home/xupp/anaconda3/bin:$PATH"

执行刚修改后的 .rc 文档：

source ~/.rc

查看 conda 版本：

conda --version

该句命令报错，找不到一个东西，那我就下载呗，结果 conda 不能用，总是调用到系统自带的 python2，Anaconda3 是使用 python3 的。

3.3.1 解决 python2 和 python3 冲突问题

查看 python 环境变量：

echo $PYTHONPATH

该句子应该是空出来的，但是显示/usr/local/lib/python2.7/dist-packages。

清空 PYTHONPATH：

unset PYTHONPATH

打开 anaconda 图形界面：

anaconda-navigator

弹出图形界面，问题解决！

3.3.2 conda 环境初始化

conda init

重新打开一个新终端就显示 base 了。

3.3.3 自由进出 base 环境

激活 base 环境：

conda activate base

退出 base 环境：

conda deactivate

设置打开新终端时，默认关闭 base 环境：

conda config --set auto_activate_base false

3.3.4 python2 的路径清空后又再次出现

考虑到 ubuntu 系统和 ros 都用到 python2，还是每次用到 Anaconda 时，执行一次：

unset PYTHONPATH

3.3.5 终端内自由切换 python2.7、python3.6、python3.8

在.rc 文件中添加：

# distinguish anaconda python with system built python
alias python="/home/zth/anaconda3/bin/python3.8"
alias python3="/usr/bin/python3.6"
alias python2="/usr/bin/python2.7"

source 一下：

source ~/.rc

4. 安装 CUDA10.2

目前最新的是 CUDA11，但是Pytorch 官网写的对于 11.1 的支持如下：

不是很 easy，所以选择 CUDA 10.2 吧。

4.1 下载 CUDA 10.2

CUDA 官网地址：网址

wget https://developer.download.nvidia.com/compute/cuda/11.1.0/local_installers/cuda_11.1.0_455.23.05_linux.run
sudo sh cuda_11.1.0_455.23.05_linux.run

安装组件的选择，取消勾选 Nvidia Driver：

4.2 配置 CUDA 环境变量

编辑 .rc 文件

sudo gedit ~/.rc

在文件末尾添加：

export PATH=/usr/local/cuda-10.2/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-10.2/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
export CUDA_HOME=/usr/local/cuda

source 一下：

source ~/.rc

检查 cuda 是否安装成功：

nvcc -V

5. 安装 cuDNN

5.1 下载 cuDNN

官网下载地址：https://developer.nvidia.com/rdp/cudnn-archive
我下载的是目前最新的 Download cuDNN v8.1.0 (January 26th, 2021), for CUDA 11.0,11.1 and 11.2 的 linux 版本。

5.2 解压缩

进入下载文件夹解压：

tar xzvf cudnn-11.2-linux-x64-v8.1.0.77.tgz

5.3 复制 cuDNN 内容到 cuda 相关文件夹内

看到网上有的是复制到 cuda 文件夹，有的是复制到 cuda-11.1 文件夹，我发现我的电脑里同时有这两个文件夹，而且内容一样，文件建立时间也完全一样，可能是同时安装的，这里我暂时不清楚原因。
复制命令：

sudo cp cuda/include/cudnn.h /usr/local/cuda-11.1/include
sudo cp cuda/lib64/libcudnn\* /usr/local/cuda-11.1/lib64

变成可执行文件：

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

6. 安装 pytorch

6.1 官网命令安装

到Pytorch 官网选择对应的系统和 cuda 版本，复制 Run this Command 中的 cuda install 命令，进入终端，

conda activate # 激活 conda 环境
unset PYTHONPATH # 清空 python2 路径

输入从官网复制的命令

我用 conda 装 pytorch 装了很久，结果失败了，不知道原因，终端输出了一大堆乱七八糟的东西，遂改用 pip 安装成功，两种方式好像并不影响使用 GPU 加速运算。

6.2 检查是否安装 GPU 版成功

以下命令可以正确输出：

nvidia-smi
nvcc -V

进入 python3.8 验证（这里我给我的 anaconda 自带的 python3.8 命名为 python，命名见前面 3.3.5 小节的叙述）：

python

输入代码：

import torch
print(torch.cuda.is_available())

如果输出的结果为 false，则表示 pytorch 无法使用 GPU 资源进行运算，需要自寻方法解决。如果输出的结果为 true，则表示 pytorch build for GPU 搭建成功。

参考博客

[1] ubuntu18.04 安装显卡 RTX3090（python3.8+nvidia Drive455.38+cuda11.1+cudnn7+pytorch1.7+torchvision0.8）

[2] Ubuntu18.04 安装 Anaconda3

[3] 在 Ubuntu18.04 上搭建 Pytorch 深度学习环境

[4] anaconda 或 python3 会自动调用系统 python2.7 下面的库

发布于 2021-05-17 00:53

yishuihan-Oliver

关注

7
点赞
踩
43

收藏

觉得还不错? 一键收藏
0
评论
ubuntu20.04 安装Anaconda3+CUDA+cudnn+Pytorch

ubuntu20.04 安装Anaconda3+CUDA+cudnn+Pytorch1、安装Anaconda：打开终端,转到自己的下载目录下（命令为：cd /home/oliver/下载），然后输入以下命令：bash Anaconda3-2020.11-Linux-x86_64.sh终端中查看Anaconda版本：conda -V 或者 conda --version2、安装NVIDIA驱动：(我选用的方法2)（方法1）查看可安装的驱动：ubuntu-drivers devices..
复制链接

扫一扫