Ubuntu24.10深度学习环境配置

前言：

NIVIDA显卡驱动已安装

查看ubuntu版本：

一、Anaconda安装

参考：超详细Ubuntu安装Anaconda步骤+Anconda常用命令_ubuntu 安装anaconda-CSDN博客

 下载地址：

Download Now | Anaconda

找到下载的文件地址，在终端打开：

执行：

bash Ana....sh  （Tab）补齐文件名

 一直按enter,直到Do you accept the license terms? [yes|no]，输入yes;

会提示确认安装位置，点击‘enter’默认即可；

初始化anaconda:

配置环境变量：
终端输入：

conda list

如果成功说明前面安装过程已经配置环境变量，如果出现conda命令找不到，需要手动设置环境变量。

找到anaconda3的地址，然后在终端输入：

sudo gedit ~/.bashrc

export PATH="/home/...自己地址/anaconda3/bin:$PATH"

保存退出，source一下：

source ~/.bashrc

再次输入conda list测试。

虚拟环境创建激活：

参考链接：Ubuntu 22.04上安装Anaconda，及 conda 的基础使用_ubuntu22.04安装anaconda-CSDN博客

终端输入：

conda create --name xpy_env(这里是虚拟环境名称) python=3.10(python版本)

 环境激活：

conda activate xpy_env

环境退出：

conda deactivate 

环境检查：

conda info --envs

二、pytorch

参考：Ubuntu22.04 安装Anaconda+虚拟环境安装Pytorch_ubuntu anaconda安装pytorch-CSDN博客

地址：PyTorch

选择符合自己要求的版本，可以进入previous versions查看。

在终端输入：

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

检查：
终端输入：

python

import torch
print(torch.version.cuda)

退出：

exit()

三、CUDA Toolkit

CUDA Toolkit 12.8 Update 1 Downloads | NVIDIA Developer

pip uninstall torch torchvision torchaudio

执行：

wget https://developer.download.nvidia.com/compute/cuda/12.8.1/local_installers/cuda_12.8.1_570.124.06_linux.run

sudo sh cuda_12.8.1_570.124.06_linux.run

执行这个上面这条命令以后会出现：

输入：accpet

出现下面窗口，通过上下键选择，按“enter”把Driver取消掉，因为我们已经安装好了驱动；

然后Instrall。

配置环境变量：

在这之前，确认以下自己的文件名及地址：

pip uninstall torch torchvision torchaudio

然后执行：

vim ~/.bashrc

按“i”插入，到最后一排根据自己的地址添加：

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

按“esc”，输入“:wq”保存退出。

source一下环境：

source ~/.bashrc

检查：

nvcc -V

四、CUDNN

pip uninstall torch torchvision torchaudio

pip uninstall torch torchvision torchaudio

执行：
 

wget https://developer.download.nvidia.com/compute/cudnn/9.10.2/local_installers/cudnn-local-repo-ubuntu2404-9.10.2_1.0-1_amd64.deb

sudo dpkg -i cudnn-local-repo-ubuntu2404-9.10.2_1.0-1_amd64.deb

sudo cp /var/cudnn-local-repo-ubuntu2404-9.10.2/cudnn-*-keyring.gpg /usr/share/keyrings/

sudo apt-get update

sudo apt-get -y install cudnn

sudo apt-get -y install cudnn-cuda-12

验证：
先检查自己的路径：

执行：

cp -r /usr/src/cudnn_samples_v9/ $HOME

cd $HOME/cudnn_samples_v9/mnistCUDNN

make clean && make

到这一部可能会提示缺少 FreeImage.h：

pip uninstall torch torchvision torchaudio

运行：

sudo apt-get install libfreeimage3 libfreeimage-dev

再执行：

make clean && make

./mnistCUDNN

成功：

五、pytorch重装

pytorch版本下错了，要重装，直接执行下面代码不行，必须先卸载pytorch.

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

卸载pytorch：

pip uninstall torch torchvision torchaudio

 再回到第一步，下载符合要求的版本。

检查：

 五、DOCKER

https://zhuanlan.zhihu.com/p/1894453286439982079