Win10 RTX3060 基于conda虚拟环境搭建pytorch-YOLOv4 （python3.6,cuda11.3）

最近在看pytorch识别相关的书，于是就买了一张RTX3060的卡跑一下机器学习和深度学习，但是没找到一套完整的RTX3060配置环境的说明，于是我就把自己配置的过程简单描述如下。

pytorch-YOLOv4的代码路径：https://github.com/Tianxiaomo/pytorch-YOLOv4

有问题欢迎讨论，谢谢！

本机的nvidia-smi信息如下：

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 472.12       Driver Version: 472.12       CUDA Version: 11.4     |
|-------------------------------+----------------------+----------------------+
| GPU  Name            TCC/WDDM | Bus-Id        Disp.A | Volatile Uncorr. ECC |

| Fan  Temp  Perf  Pwr:Usage/Cap|         Memory-Usage | GPU-Util  Compute M. |
|                               |                      |               MIG M. |
|===============================+======================+======================|
|   0  NVIDIA GeForce ... WDDM  | 00000000:06:00.0  On |                  N/A |
|  0%   38C    P8    12W / 170W |   1043MiB / 12288MiB |      7%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

nvcc -V (cuda 版本为11.3):

nvcc -V
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Mar_21_19:24:09_Pacific_Daylight_Time_2021
Cuda compilation tools, release 11.3, V11.3.58
Build cuda_11.3.r11.3/compiler.29745058_0

安装python 3.6 虚拟环境：

conda create -n py36 python=3.6

安装pytorch（用cudatoolkit 11.0， pytorch 1.7.0）：

请参考Previous PyTorch Versions | PyTorch ，官网中pytorch 1.9.0 与cudatoolkit 11.3的版本是cpu版本，识别时会有问题，经过多个pytorch的版本比较，最终找到了最适合的版本为pytorch1.7。RTX3060 显卡支持的最小cuda版本是11.0，如果用cuda 10.2时会出现显卡无法使用的情况。（为了使3060能正常使用，我查阅了很多资料，经过多次重新配置虚拟环境，才找到这个匹配的环境套装）

# CUDA 11.0
conda install pytorch==1.7.0 torchvision==0.8.0 torchaudio==0.7.0 cudatoolkit=11.0 -c pytorch

python opyouencv 安装：

pip安装指令：

pip install opencv-python

下载离线包安装：

​​​​​​​

链接：https://pan.baidu.com/s/10i_2c1YRFDeJUqDmQAGwYA 
提取码：6jp7 

下载完成后考到目录下，执行 pip install opencv_python-4.4.0-cp36-cp36m-win_amd64.whl即可。