安装pytorch时NVIDIA和CUDA版本问题
电脑上的NVIDIA在下载pytorch时没有对应CUDA版本,比如我的本来是9.2,pytorch官网只有10.2和11.1。可以更新自己的NVIDIA版本,再下载对应CUDA版本。
1GeForce驱动下载:https://www.nvidia.cn/geforce/drivers/
输入驱动信息,下载最接近的版本号
2. pytorch官网选择对应信息下载
参考:https://blog.csdn.net/Mozen_/article/details/109274527
https://www.zhihu.com/question/472220337