一、配置环境  
1，查看CUDA版本  
在搜索框输入NVIDIA，点击NVIDIA控制面板  
选中“管理3D设置”，点击下方的“系统信息”  
点击“组件”按钮，查看CUDA版本

![Exported image](Exported%20image%2020260425172256-0.png)  

2、根据所需版本下载，安装CUDA  
[https://developer.nvidia.com/cuda-toolkit-archive](https://developer.nvidia.com/cuda-toolkit-archive)  
3、下载cudnn  
[https://developer.nvidia.com/cudnn](https://developer.nvidia.com/cudnn)  
安装方法：解压，覆盖CUDA的文件  
4、安装pytorch【必须在python环境下安装】  
[https://pytorch.org/get-started/locally/](https://pytorch.org/get-started/locally/)  
5、运行文件检查

|   |   |
|---|---|
|参考操作|[https://blog.csdn.net/anmin8888/article/details/127910084](https://blog.csdn.net/anmin8888/article/details/127910084)|