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
安装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
创建虚拟变量,指定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