![fdc2f5a396fd181e02c08f1656840845.png](https://i-blog.csdnimg.cn/blog_migrate/6e0be4829acf966a63b1a36721ed7339.jpeg)
![f7446a982ad450f099c766b25afd7cbe.png](https://i-blog.csdnimg.cn/blog_migrate/038cd1163abf0f7383c22029f596e21a.png)
只选择CUDA 其他选项不选
![cc6900bf43a108b705ccb8332e679a7e.png](https://i-blog.csdnimg.cn/blog_migrate/4474386374d9ac328ae04479cbe7da86.jpeg)
安装位置可以自己定 例如
D:cudaNVIDIA CorporationCUDA Samples
D:cudaNVIDIA GPU Computing ToolkitCUDAv10.1
![d8567a8e730572a885199a1bc350f4b0.png](https://i-blog.csdnimg.cn/blog_migrate/a4e531e96b8317354ccff68b92cd0b98.jpeg)
然后在安装CUDNN
![d843f13c1c1baed99bb908b320e27de9.png](https://i-blog.csdnimg.cn/blog_migrate/9f950d123f7729e700e5e4de70d8bede.jpeg)
将其解压后 将对应文件夹做合并!!! 不要直接估值粘贴
然后设置环境变量
![e169c7d829745eb9f0e6accd27d20b41.png](https://i-blog.csdnimg.cn/blog_migrate/352d0d1846bca4c7208db004c7acbef4.jpeg)
![70f693ccf115bcdfe991e5aa27b9bf44.png](https://i-blog.csdnimg.cn/blog_migrate/071aca86887c784462b6fe7f8606e2f6.jpeg)
![4417a3ba7c9e9e40d83e7a76402f65e1.png](https://i-blog.csdnimg.cn/blog_migrate/4ec2e54d36d873b803b7fc3bf3995c50.jpeg)
测试cuda是否安装成功
输入 nvcc -V
![6c353dee5e327174cf22f2ed1d1c54c4.png](https://i-blog.csdnimg.cn/blog_migrate/05794f221864923b618071e93ad189ca.jpeg)
然后将路径转到
D:cudaNVIDIA GPU Computing ToolkitCUDAv10.1extrasdemo_suite
![eef6d13fd27aa9bad927ea8942671fcd.png](https://i-blog.csdnimg.cn/blog_migrate/de42fd6c510bc6098fd78440efba0538.jpeg)
之后输入 .banwidthTest.exe
和 .deviceQuery.exe
出现Result = PASS 就是安装成功
![7717746f07bd67ba72736c539111b289.png](https://i-blog.csdnimg.cn/blog_migrate/dad6969d9e597df7ebfa8c6879da0ef3.jpeg)
![d69b63833519ac98c67710ed42e0b43f.png](https://i-blog.csdnimg.cn/blog_migrate/f2a8ec92face381d1dd96df68d3e8b1e.jpeg)
离线安装pytorch
pytorch安装包download.pytorch.org选择对应cuda 的版本
下载完后使用 pip安装
切记安装在确定的环境中
将路径切到安装所在位置 直接pip
![d408dade00c8e972aa17831e84d0e395.png](https://i-blog.csdnimg.cn/blog_migrate/c23df18858f81de10924d009057e4689.jpeg)
测试是否安装pytorch成功
import torch
print(torch.__version__)
判断pytorch是否支持GPU加速
print(torch.cuda.is_available())
输出为:True,即安装成功