出现如下问题:
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.csdn.net/qq_39938666/article/details/86611474
https://blog.csdn.net/weixin_41786536/article/details/89497006
https://blog.csdn.net/SwordMaster_Ice/article/details/89134380