ubuntu24.04安装模型训练环境

  1. 安装英伟达驱动
    应用程序-> 附加驱动 → 选择驱动版本 → 应用更改(需要选择点击两遍)->重启



    控制台输入nvidia-smi

  2. 安装cuda
    根据上面图片,比如我这个电脑安装的550的驱动,对应的版本就是12.4,去官网下载12.4版本。官网地址


    选择对应的版本

    根据顺序执行下方的命令,大概会在最后一条报错

解决方案:

  • 打开镜像文件并修改
    sudo vim /etc/apt/sources.list.d/ubuntu.sources
  • 向sources文件中添加如下内容
    Types: deb
    URIs: Index of /ubuntu
    Suites: lunar
    Components: universe
    Signed-By: /usr/share/keyrings/ubuntu-archive-keyring.gpg
  • 更新
    sudo apt-get update
  • 重新安装
    sudo apt install cuda -y

cuda环境变量配置:

sudo nano ~/.bashrc
  • 将以下内容添加进文件最后
export PATH=/usr/local/cuda-12.0/bin${PATH:+:${PATH}}
export LD_LIBRARY_PATH=/usr/local/cuda-12.0/lib64${LD_LIBRARY_PATH:+:${LD_LIBRARY_PATH}}
  • 注意修改上面的版本号为自己下载的版本
  • 保存退出后(Ctrl+x),更新一下环境变量:
source ~/.bashrc

这时候在执行 nvcc -V 就能够显示cuda版本了。

  1. 安装cudnn
    cuDNN是对CUDA的一些补充,下载对应CUDA12.4的cuDNN压缩包(这一步可能需要注册英伟达账号)。解压之后得到cuda目录,cuda目录下面有include和lib64两个子目录,将这两个目录下面的所有文件拷贝到CUDA 12.4安装路径对应的目录下面即可。
    cuDNN下载地址
tar -xvf cudnn** 
cd cudnn-linux-x86_64-8.9.3.28_cuda12-archive
#以下是安装命令     
sudo cp -r lib/* /usr/local/cuda-12.4(自己检查具体的版本修改路径)/lib64/
sudo cp -r include/* /usr/local/cuda-12.4(自己检查具体的版本修改路径)/include/

#为更改读取权限:
sudo chmod a+r /usr/local/cuda-12.4(自己检查具体的版本修改路径)/include/cudnn*
sudo chmod a+r /usr/local/cuda-12.4(自己检查具体的版本修改路径)/lib64/libcudnn*

检查cudnn是否安装成功

cat /usr/local/cuda-12.0/include/cudnn_version.h | grep CUDNN_MAJOR -A 2

  1. 安装conda
    conda下载地址
    安装 Anaconda
bash Anaconda3-2024.10-1-Linux-x86_64.sh

输入yes 回车直到安装成功
注意事项:


conda init初始化是no 需要输入yes 会设置环境变量

如果没有设置的话,也不要急,可以手动初始化

source ~/anaconda3/bin/activate

执行完成后进入base环境
图片

conda init

执行完成后最后一行如下:
图片
关闭终端重新打开

cat ~/.bashrc
conda --version

conda就安装完毕了

其他的conda命令

#创建虚拟环境
conda create -n your_env_name python=X.X(3.6、3.7等)
 
#激活虚拟环境
source activate your_env_name(虚拟环境名称)
 
#退出虚拟环境
source deactivate your_env_name(虚拟环境名称)
 
#删除虚拟环境
conda remove -n your_env_name(虚拟环境名称) --all
 
#查看安装了哪些包
conda list
 
#安装包
conda install package_name(包名)
conda install scrapy==1.3 # 安装指定版本的包
conda install -n 环境名 包名 # 在conda指定的某个环境中安装包
 
#查看当前存在哪些虚拟环境
conda env list 
#或 
conda info -e
#或
conda info --envs
 
#检查更新当前conda
conda update conda
 
#更新anaconda
conda update anaconda
 
#更新所有库
conda update --all
 
#更新python
conda update python
  1. 使用conda安装yolov8
  • 创建虚拟环境
    图片
  • 激活虚拟环境
    图片
  • 安装依赖
    图片
  • 安装pytorch
    图片
  • 安装yolo
pip install ultralytics
  • 测试YOLOv8 是否安装成功
yolo predict model=yolov8n.pt source='https://ultralytics.com/images/bus.jpg

21 Likes

好帖 :tieba_095: :tieba_095: :tieba_095:

5 Likes

谢谢 :grinning: :grinning: :grinning:

2 Likes

先mack一下 :grinning:

2 Likes

感谢大佬教程

2 Likes

pip安装了几个cuda的python包好像也可以训练,然后只安装了英韦达驱动,不安装cuda toolkit也可以吗

1 Like

好像是可以的 cuda主要是用于GPU的使用

感谢您的分享:rose:

1 Like

没必要装 CUDA 相关的其实,conda 会自己处理 CUDA 相关包(装 torch 之类的时候),不需要全局装。

2 Likes

感谢大佬指导,主要是刚学不久,没啥经验

1 Like

强啊 这个不错

1 Like

占个位置,后面来刷刷看

1 Like

我也是这样想的,现在cudnn不用手动装了

1 Like

佬环境都搭好了,不再分享几个训练的案例,给L们开开眼么 :joy:

1 Like

哈哈哈哈,这个yolo库就有现场的,你可以先试试,不行可以问我

这个还真不太清楚,查的资料可能有点久了,就装上了

感谢分享!!

1 Like

感谢分享 !!!

1 Like

感谢佬友分享

1 Like

客气了 :bili_029: