本文用来预防pytorch出问题需要重装:(
一、下载安装对应版本CUDA和cudnn
(之前下载的,现在忘记记录在哪里了,找到记录再放,或者去参看其他博客)
版本选择参考:
GPU版本的PyTorch安装与环境配置_ke996的博客-CSDN博客_安装gpu版本的pytorch
安装结果
![](https://i-blog.csdnimg.cn/blog_migrate/2c78582922239c089e07f35ff4525b40.png)
二、下载torch,并安装
1.下载地址https://download.pytorch.org/whl/torch_stable.html
2.版本对应关系
(1)CUDA & PyTorch
![](https://i-blog.csdnimg.cn/blog_migrate/aebf7d8e58ab268c19f7e8f43b09152a.png)
(2)选择python对应版本,或者安装对应版本python(后来因参考代码需要改下1.8.0版本)
![](https://i-blog.csdnimg.cn/blog_migrate/a04e269c592571403cfbd103594c642e.png)
3.安装
cmd到下载目录下,使用pip install
![](https://i-blog.csdnimg.cn/blog_migrate/c0bbe78a1d5052ec2aeac7eace930c13.png)
三、测试使用
![](https://i-blog.csdnimg.cn/blog_migrate/9b1b796495fd75a0e9e9e6cf0ee5c8a4.png)
![](https://i-blog.csdnimg.cn/blog_migrate/133dbf76d000a3a3da28fb37df318099.png)