Pytorch正确安装但是无法启用CUDA

情况

  • 安装pytorch版本为pytorch1.8.1+cu11.1,win10,运行时无法调用CUDA,显示初始化失败。
Traceback (most recent call last):
  File "anaconda\envs\dp_tf\lib\site-packages\torch\cuda\__init__.py", line 170, in _lazy_init
    torch._C._cuda_init()
RuntimeError: CUDA driver initialization failed, you might not have a CUDA gpu.
  • 电脑已安装CUDA11.1,但是Pytorch无法调用

cuda

  • 输入torch.backends.cudnn.enabled 其输出Out[]: True
  • 输入torch.cuda.is_available()其输出Out[]: False

解决办法

  • 电脑的显卡之前更换为老版本驱动,在更新为最新版本的驱动后,即可正常调用

  • Nvidia Geforce Experience中,更新显卡驱动,即可正常调用。

你可能感兴趣的:(学习笔记,机器学习,cuda,gpu)