Ubuntu20.04配置深度学习+yolov5+yolov7环境

目录

一、换国内源

二、安装gpu驱动

三、Anaconda安装

四、pytorch-cpu版本安装

五、cuda安装

六、cudnn安装

七、pytorch-gpu安装

八、tensorflow安装

九、yolov5配置

十、yolov7配置

---

我的配置：（配置不同软件版本要求不同，建议查一查先）

ubuntu20.04

RTX4070ti

NVIDIA-SMI 550.78

CUDA Version: 12.4  (最高支持12.4但我装的12.1)

一、换国内源

换好后更新一下

sudo apt update
sudo apt upgrade

二、安装gpu驱动

1.禁用nouveau

sudo gedit /etc/modprobe.d/blacklist.conf

文件最后一行添加

blacklist nouveau

 保存后终端运行

sudo update-initramfs -u

2.卸载旧的驱动（第一次安装驱动，可省略）

3.安装gcc

sudo apt install gcc

4.下载安装对应版本的gpu驱动及验证

具体参考B站Ubuntu安装NVIDIA显卡驱动

三、Anaconda安装

1.安装Anaconda

下载对应版本anaconda下载链接 64bit（x86）

cd到下在文件所在目录进行安装

bash Anaconda3-2024.02-1-Linux-x86_64.sh

然后一直按回车键，直到出现“Do you accept the license terms”, 输入“yes” 

出现“Anaconda3 will now be installed into this location: /home/xxx/anaconda3”，直接按回车键安装在该目录中。

出现“Do you wish the installer to initialize Anaconda3 by running condo init?”, 输入“no”。

这时anaconda已经基本安装完成了。

2.区分Ubuntu自带python和anaconda中的python

Linux中会自带python, Ubuntu20.04中自带python3,已经没有python2了。这时不区分不影响anaconda的正常使用，但pytorch和tensorflow对这Linux中自带的python3和anaconda中的python的使用需求不同，后期需要进行两者的切换。

cd到根目录

cd ~

安装vim

sudo apt install vim

用vim打开.bashrc文件，并进行修改。

sudo vim .bashrc \\利用vim打开.bashrc文件

输入小写字母i，进行vim的编辑模式。此时可以在.bashrc文件进行编辑了，在文件最后加入以下命令： 

alias python3="/usr/bin/python3.8"\\给系统自带的python起一个别名，就叫python3
export PATH="/home/xxx/anaconda3/bin:$PATH"\\anaconda3中的python

 然后按下esc键，再输入:wq!后按回车。

继续在终端中对刚才修改的. bashrc文件执行以下。

source .bashrc

python \\本条命令应该启动anaconda3中的python
exit() \\退出
python3 \\本条命令应该启动系统的python
exit() \\退出

3.创建D