C:\Users\ASUS\anaconda3\lib\site-packages\torch\cuda\__init__.py:52: UserWarning: CUDA initialization: The NVIDIA driver on your system is too old (found version 9010).
Please update your GPU driver by downloading and installing a new version from the URL: http://www.nvidia.com/Download/index.aspx
Alternatively, go to: https://pytorch.org to install a PyTorch version that has been compiled with your version of the CUDA driver. (Triggered internally at ..\c10\cuda\CUDAFunctions.cpp:100.)
return torch._C._cuda_getDeviceCount() > 0字符串
的数据
我的系统：Windows 10 Nvidia GeForce 940 MX
1条答案
按热度按时间hujrc8aj1#
使用pytorch设置ML环境的快速检查表（从业者的建议-按照检查表中的顺序执行，* 绑定很重要 *）：
1.获取一些good GPU（*940 MX不是首选 *）
1.安装NVIDIA drivers（添加了v381的链接，一旦安装，在Windows平台上，你会得到更新通知！）
1.安装CUDA Toolkit
1.安装CuDNN
1.安装Anaconda
1.创建conda环境
1.安装PyTorch
你可以在互联网上找到大量的博客，这些博客将详细地指导你完成这些步骤。