CUDA initialization: Found no NVIDIA driver on your system. Please check that you have an NVIDIA GPU and installed a driver from http://www.nvidia.com/Download/index.aspx (Triggered internally at /opt/conda/conda-bld/pytorch_1603729006826/work/c10/cuda/CUDAFunctions.cpp:100.) return torch._C._cuda_getDeviceCount() > 0 False
时间: 2023-10-09 12:13:49 浏览: 890
TI-TCA6424A.pdf
这个错误提示是因为你的系统中没有安装NVIDIA驱动。你需要到NVIDIA官网下载并安装适合你所使用的GPU型号的驱动程序。下载链接为:http://www.nvidia.com/Download/index.aspx。安装完成后,重启你的计算机并重新运行你的代码。如果还有问题,可以尝试更新你的CUDA版本或者PyTorch版本。
阅读全文