现象:运行了torch.cuda.is_available()后显示
Error 500: named symbol not found (Triggered internally at /opt/conda/conda-bld/pytorch_1695392020201/work/c10/cuda/CUDAFunctions.cpp:108.)
return torch._C._cuda_getDeviceCount() > 0
False
网上搜了大部分都说是重启解决,再者就是pytorch的版本和cuda版本不兼容,但我的image直接下载的就是docker hub中的pytorch/pytorch,不应该会出现版本不兼容的问题。
image地址:https://hub.docker.com/r/pytorch/pytorch
接下来再去寻找其他的方式,最后试下来是要重装显卡驱动,回退版本(也有说只需重装就行,无需回退版本)
如需回退版本,可在官方高级驱动搜索 | NVIDIA中寻找之前的版本