Ubuntu18.04下安装pytorch步骤&多cuda版本共存（啰嗦版）

九三久久

已于 2023-04-07 23:31:31 修改

阅读量5.7k

点赞数 14

文章标签： pytorch linux ubuntu

于 2023-04-07 23:29:47 首次发布

本文链接：https://blog.youkuaiyun.com/weixin_54489145/article/details/129940366

版权

本文详细介绍了在Ubuntu系统上安装CUDA、cuDNN以及PyTorch-GPU的步骤，包括显卡驱动的安装、CUDA和cuDNN的版本选择与安装，以及如何通过Anaconda创建虚拟环境安装PyTorch。文章强调了安装过程中要注意的版本兼容性、驱动安装方法，以及如何检查安装是否成功。此外，还提供了卸载CUDA的参考链接和多版本CUDA环境的配置方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

前言

1.知识补充

【机器学习】显卡、GPU、NVIDIA、CUDA、cuDNN（搬运：要点如下，详细可看链接）

加 * 非重要内容，视情况执行。

显卡：即显示卡，全称显示接口卡，是计算机最基本配置、最重要的配件之一（就像联网需网卡，数据显示在屏幕需显卡）。显卡是由GPU、显存等等组成的。

GPU：图形处理器，一般焊接在显卡上的。GPU是显卡上的一块芯片，就像CPU是主板上的一块芯片。GPU功能强大，只用于图形处理太浪费，NVIDIA公司提出CUDA的概念，通用并行计算架构，是一种运算平台，更加方便利用GPU强大的计算能力（并非所有GPU支持CUDA）。

CUDA：通用并行计算架构，该架构使GPU能够解决复杂的计算问题。（加加速-我的理解）

cuDNN：（只是cuda的扩展工具而已）是CUDA在深度学习方面的应用。使得CUDA能够针对性的应用于加速深度神经网络。

2.安装说明

1.如果没有显卡，直接 1 与 3 即可，途中标颜色即建议步骤，只有 2.2.2 较为复杂；

2.所有安装包，都可以备份 U 盘，下次可以直接使用；

3.笔者认为cuDNN 只是 cuda 扩展工具，下载了对应 cuda的 cuDNN，复制 cuDNN 文件到 cuda 即可。所以安装步骤放 cuda 中的一个小点，没必要作为一个大标题。

1、Anaconda的下载安装

搬运：Ubuntu18.04下安装Pytorch-GPU（超详细自己安装全过程）

1.1 下载

最新版本：Anaconda官网下载

历史版本：Anaconda历史版本

下载过慢：Anaconda清华源下载

（其实可以将安装文件保存 U盘，下次安装可以直接使用_cuda/cuDNN 也是如此）

1.2 安装

1. 终端运行下载 .sh 文件：

bash Anaconda3-（下载文件 Tab 补全即可）-Linux-x86_64.sh

2.过程

Welcome to Anaconda3 5.0.1
 
In order to continue the installation process, please review the license
agreement.
Please, press ENTER to continue
>>> 按回车
#然后一直按回车到协议完毕
#出现：
Do you accept the license terms? [yes|no]
>>>输入yes
#下面就是问你安装目录，建议就是默认的安装路径，直接按回车
Anaconda3 will now be installed into this location:
/home/mayunteng/anaconda3
 
  - Press ENTER to confirm the location
  - Press CTRL-C to abort the installation
  - Or specify a different location below
 
[/home/mayunteng/anaconda3] >>> 按回车
#接下来就是等待安装完成
#注意安装完成以后会询问你是否把anaconda3的路径加到环境变量里去，一定要选yes,一定要选yes,一定要选yes。

*1.3 添加环境变量（1.2默认则不需要执行）

若 1.2 默认路径添加环境变量，新开终端出现会出现 (base)，表明已经成功添加环境路径，则无需执行以下代码。

sudo gedit ~/.bashrc

# 在文件末尾处添加以下语句
export PATH=/home/XXX/anaconda3/bin:$PATH    XXX为自己的用户名

source ~/.bashrc

终端输入python，可以看到有Anaconda出现，安装成功。

2、安装cuda（cuDNN是配套的）

在安装之前你要先查看你的电脑是否支持GPU运算,否则你也不用安装了。

2.1 NVIDIA驱动安装

平常使用 2.1.1 两行命令安装就可以了，实在安装有问题可以尝试 2.1.2。

2.1.1 命令自动安装

Ubuntu18.04中自带了NVIDIA驱动，但没有完整安装，使用命令ubuntu-drivers devices（若不显示，sudo apt update更新源）可查看当前的设备和驱动。

ubuntu-drivers devices

sudo ubuntu-drivers autoinstall #安装所有驱动 
sudo apt install nvidia-440 #只安装其中一个驱动 

sudo nvidia-smi #验证是否成功

*2.1.2使用官方的NVIDIA驱动进行手动安装（稳定、靠谱）

*下载 cuDNN 没注意，显卡驱动自动降版后，自动安装装不上，使用手动安装成功解决。

（纯纯的搬运工，主要真的有用，怕作者哪天删除了）

参考链接：Ubuntu18-22.04安装和干净卸载nvidia显卡驱动——超详细、最简单

最低0.47元/天解锁文章