使用GPU训练网络模型+跑yolo（包含Anconda的安装）_用gpu运行模型 需要安装哪些软件

OK，上次写博客还是在大三，现在研一，导师让用GPU来训练网络模型，并运行yolo。

安装显卡驱动 + cuda驱动

使用nvidia-smi命令查看自己的显卡驱动，然后安装适合自己的cuda Driver version
cuda 驱动下载

进入pytorch官网安装pytorch

选择自己的平台和cuda 版本，然后得到一个命令，这个命令我们不着急使用

##安装pytouch 2.0.1 cuda 10.1 注意cuda版本要与cuda驱动一致
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch -c defaults -c numba/label/dev
1
2

安装Anconda并创建Anconda虚拟环境

Anconda 下载地址

环境变量：

D:\Program Files\Anaconda3
D:\Program Files\Anaconda3\Scripts 
D:\Program Files\Anaconda3\Library\mingw-w64\bin
D:\Program Files\Anaconda3\Library\usr\bin 
D:\Program Files\Anaconda3\Library\bin
1
2
3
4
5

创建虚拟变量，指定python3.8，激活虚拟环境后再执行pytorch官网代码

1、canda 创建环境
conda create -n pytorch_gpu_venv python=3.8

2、激活环境变量
activate pytorch_gpu_venv

3、安装pytouch 2.0.1 cuda 10.1 注意cuda版本要与cuda驱动一致
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch -c defaults -c numba/label/dev

1
2
3
4
5
6
7
8
9