Pytorch GPU CUDA版本匹配

出现如下问题：

Found GPU0 TITAN V which requires CUDA_VERSION >= 9000 for optimal performance and fast startup time, but your PyTorch was compiled with CUDA_VERSION 8000. Please install the correct PyTorch binary using instructions from http://pytorch.org warnings.warn(incorrect_binary_warn % (d, name, 9000, CUDA_VERSION))

直接用pip install torch==0.3.1安装的是cuda8.0的，使用如下命令：

pip install http://download.pytorch.org/whl/cu90/torch-0.3.1-cp35-cp35m-linux_x86_64.whl

 

补充：

在另一台服务器上跑的时候提示需要9.0以上的CUDA，显卡为RTX 2080Ti，结果报错 torch.backends.cudnn.CuDNNError: 8: b'CUDNN_STATUS_EXECUTION_FAILED' ，实际上需要10.0以上的CUDA

 

 

参考链接：

https://blog.youkuaiyun.com/qq_39938666/article/details/86611474

https://blog.youkuaiyun.com/weixin_41786536/article/details/89497006

https://blog.youkuaiyun.com/SwordMaster_Ice/article/details/89134380

https://cloud.tencent.com/developer/article/1476954