YOLOv5 - Win10 + RTX3070 的 PyTorch 安装指南（ASUS 天选 air）

目录

前言

1. 选取合适的软件及版本

2. 安装VS2017

3. 安装CUDA

4. 安装设置 Anaconda

5. 安装PyTorch

6. 安装YOLOv5

7. 测试YOLOv5

小结

---

前言

华硕推出了RTX3070的天选系列，好不容易抢了台ASUS 天选air，就想试试RTX3070的AI性能怎么样，这货有5120个CUDA核心，8GB GDDR6显存，应该比较牛吧！

YOLOv5最近也比较火，就拿它来试试。正好整理一下相关的软件，附上链接。

1. 选取合适的软件及版本

YOLOv5 需要安装PyTorch，登录PyTorch的网站，找到最新的稳定版本为1.7.1，目前为止，支持的最高的CUDA版本为11.0。

我安装在windows系统，采用Conda安装，选择Python 3.8，RTX3070是Ampere构架，支持的最低的CUDA版本为11.0（大家可以关注一下，随着厂家的后续更新，可以选择支持的最新版本）。

注意：我开始选择的是CUDA Toolkit 11.0 Update1 (Aug 2020)，发现不能正确识别RTX3070。

后来重新安装CUDA Toolkit 11.2，就可以正确识别RTX3070了，看来CUDA的小版本对PyTorch的影响不是太大，不是必须安装CUDA 11.0版本。 

选择对应的cuDNN 8.1.0，CUDA Driver Version应该是最低要求，我安装了最新的 r461。

仔细研究发现，RTX3070的计算能力是8.6，11.0的CUDA里面没有RTX30xx的信息，11.2的CUDA里面有RTX30xx的信息：{0x86，"Ampere"}。

C:\ProgramData\NVIDIA Corporation\CUDA Samples\v11.2\common\inc\helper_cuda.h

CUDA需要有一个Compiler，VS2017/VS2019都支持。我选的是Visual Studio Community 2017 社区版（免费版），

Note: There's a compilation issue in several Visual Studio 2019 versions since 16.7.1, so please make sure your Visual Studio 2019 version is not in 16.7.1 ~ 16.7.5

 

2. 安装VS2017

下载并安装