Pytorch(GPU版本)环境构建

在Windows下完成GPU版Pytorch的安装。

1.确认机器的显卡已定是Nvidia的。这个默认系统已经安装了Nvidia驱动

(AMD的显卡可能会出问题，所以如果是AMD的显卡建议直接安装CPU版的)

2.安装CUDA
这是GPU版Pytorch所必须的。Pytorch其实是对CUDA的封装，让编程更加简单，但是丧失了效率，将来我们所要学的就是将PYTORCH训练好的模型转成CUDA和TensorRT。

进入“https://developer.nvidia.com/cuda-downloads”，选择对应的系统配置下载cuda.xxxxxx.exe文件

类似一般性的windows应用的安装方式安装即可。
现在的问题是如果按照这个流程下来的话安装的cuda应该是11版本，建议直接从我已经上传好的文件下载.
百度云：链接：

https://pan.baidu.com/s/1mWH7BZ1Wc1bAKp6FDBVwWw

提取码：bztw
验证是否已安装：打开cmd,输入：nvcc -V，输出：