Ubuntu 20.04安装显卡驱动、CUDA、Miniconda和Pytorch

发布于:2025-04-02 ⋅ 阅读:(27) ⋅ 点赞:(0)

https://blog.csdn.net/sdbyp/article/details/139853774?utm_medium=distribute.pc_relevant.none-task-blog-2defaultbaidujs_baidulandingword~default-0-139853774-blog-139606986.235v43pc_blog_bottom_relevance_base3&spm=1001.2101.3001.4242.1&utm_relevant_index=3

其中:显卡驱动安装 可以参考https://editor.csdn.net/md/?articleId=146912421
查看 系统版本

在这里插入图片描述

CUDA Toolkit 12.6 Update 2 Downloads

在这里插入图片描述

在这里插入图片描述

在这里插入图片描述
完成
在这里插入图片描述

2.4 环境变量配置

以vim方式打开配置文件

sudo vim ~/.bashrc

在文件尾中加入以下两行:

export PATH="/usr/local/cuda-12.6/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-12.6/lib64:$LD_LIBRARY_PATH"

在这里插入图片描述

更新环境变量

source ~/.bashrc
2.5 检测CUDA是否安装成功
nvcc -V

在这里插入图片描述

命令行显示如上信息,表明安装成功!大功告成~

三、Miniconda

Anaconda简介:Anaconda 是一个用于科学计算的 Python 发行版,支持 Linux, Mac, Windows, 包含了众多流行的科学计算、数据分析的 Python 包。


Anaconda和Miniconda的区别:Anaconda软件包大概500MB,安装后大概3-4G,base环境包括了conda管理器、Pyhon编译器、常用的包和Spyder IDE等;Miniconda软件包大概50MB,安装后大概300MB,base环境选择只包括conda管理器和Python编译器,其他的包需要通过conda命令安装。
3.1 下载Miniconda

官网链接:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/

选择版本:
https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/Miniconda3-py39_4.9.2-Linux-x86_64.sh
在这里插入图片描述

3.2 安装Miniconda

(1)开始安装。输入以下命令后,敲击回车键

sudo bash Miniconda3-py38_23.11.0-1-Linux-x86_64.sh

在这里插入图片描述

(2)接受条款。按着Enter不松,直到出现条款,输入yes:

在这里插入图片描述

(3) 输入安装路径后敲击回车键。最好指定安装路径:/usr/local/miniconda

我的忘记输入了,按默认安装的
在这里插入图片描述

(4)最后敲击回车键
在这里插入图片描述

3.3 配置环境
(1)配置环境变量。修改/etc/profile文件,来配置环境,添加一行(根据自己的安装路径添加)

export PATH=/usr/local/miniconda/bin:$PATH

我的忘记指定了,按默认安装路径 
export PATH=/root/miniconda3/bin:$PATH

在这里插入图片描述

(2)配置生效。

source /etc/profile

(3) 验证是否安装成功

conda -V

在这里插入图片描述

若显示版本号,表明miniconda已安装成功!

然后就可以使用conda 来创建python 虚拟空间了

conda create -n ai_pyenv_3.12 python=3.12 -y \
--override-channels \
-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main \
-c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free
# To activate this environment, use
#
#     $ conda activate ai_pyenv_3.12
#
# To deactivate an active environment, use
#
#     $ conda deactivate

在这里插入图片描述
在这里插入图片描述

在这里插入图片描述

————————————————

五、关于安装cuDNN(这个坑以后再补)

cuDNN是一个基于CUDA的深度学习GPU加速库,可以提高深度学习模型在GPU上的训练速度。cuDNN不是必须安装的,但是一般会采用这个加速库。
部分深度学习框架,如PyTorch,**会自带预先编译的cuDNN**,无需手动安装。
后面我还需要安装Pytorch,所以cuDNN就没安装,需要安装的可以参考下面信息,但是不完整。

版本查看页面:https://developer.nvidia.com/cudnn-downloads
在这里插入图片描述

下载CUDA 12对应的cuDNN版本,这里我们选择8.9.0版本。点击该版本,显示如下,我们选择红框圈中的版本。
在这里插入图片描述

step4:重点!重点!重点!

大部分人在这一部就找不到libcudnn8、libcudnn8-dev、libcudnn8-samples文件,导致无法安装成功,此处如果是安装在默认路径,可以cd至/var/cudnn-local-repo-ubuntu2204-8.9.0.131/目录下查看文件:

cd /var/cudnn-local-repo-ubuntu2204-8.6.0.163/

在该目录下就会看见这三个文件,使用下方指令分别进行安装:

sudo dpkg -i libcudnn8_8.6.0.163-1+cuda11.8_amd64.deb
sudo dpkg -i libcudnn8-dev_8.6.0.163-1+cuda11.8_amd64.deb
sudo dpkg -i libcudnn8-samples_8.6.0.163-1+cuda11.8_amd64.deb

终端打印如下:
在这里插入图片描述

step5:查看cuDNN安装结果:

cat /usr/include/x86_64-linux-gnu/cudnn_version_v8.h | grep CUDNN_MAJOR -A 2

打印版本号,安装成功了。

至此安装完成。
————————————————

四、安装Pytorch

4.1 选择合适Pytorch版本

链接:https://pytorch.org/get-started/previous-versions/

注意:执行nvidia-smi命令,确认 cuda 版本需要大于等于pytorch安装的对应版本
在这里插入图片描述

打开终端(ctrl+alt+t),输入上述命令,即完不安装。
我的:


​pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu126

其他:

pip install torch==2.5.1 torchvision==0.20.1 torchaudio==2.5.1 --index-url https://download.pytorch.org/whl/cu121
pytorch 下载太慢解决 见下文

检查安装是否成功

python -c "import torch; print(torch.cuda.is_available())"

输出True 这说明 GPU版本的pytorch​安装成功
在这里插入图片描述
前面的 (myenv) 是python的虚拟环境:参见 https://blog.csdn.net/nalanxiaoxiao2011/article/details/146916168

方式二、编写 Python 脚本验证

在文本编辑器中创建一个新的 Python 文件,例如 test_pytorch.py。
在文件中输入以下代码:

python
import torch

try:
    import torch
    print("PyTorch 安装成功!")
    print(f"PyTorch 版本: {torch.__version__}")
    print(f"CUDA 是否可用: {torch.cuda.is_available()}")
except ImportError:
    print("PyTorch 未安装成功。")

保存文件,然后在命令行终端中导航到该文件所在的目录,运行 python test_pytorch.py 命令。根据输出结果判断 pytorch 是否安装成功以及相关信息。

在这里插入图片描述