【BUG解决】pytorch识别不到cuda问题 torch.cuda.is_available() False问题解决

明月下

已于 2024-12-02 21:59:29 修改

阅读量2.4k

点赞数 2

CC 4.0 BY-SA版权

分类专栏：环境搭建和BUG解决文章标签： pytorch 人工智能 python

于 2023-09-15 17:01:04 首次发布

本文链接：https://blog.csdn.net/qq_45384162/article/details/132907516

环境搭建和BUG解决专栏收录该内容

6 篇文章

订阅专栏

博主此前使用pytorch.cuda正常，两周未用后报错。虽nivida470与cuda11.4、cuda11.4与pytorch11.3版本一致，环境变量设置也正确，但尝试卸载重装、更新nvidia - modprobe等均未解决，最终通过重启解决问题。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

BUG

之前都能正常用pytoch.cuda,两个星期没用，忽然就报错了。
报错如下：

UserWarning: CUDA initialization: CUDA unknown error - this may be due to an incorrectly set up environment, e.g. changing env variable CUDA_VISIBLE_DEVICES after program start. Setting the available devices to be zero. (Triggered internally at  /opt/conda/conda-bld/pytorch_1614378083779/work/c10/cuda/CUDAFunctions.cpp:109.)
  return torch._C._cuda_getDeviceCount() > 0

并且

print(torch.cuda.is_available())
#输出
False

但是我一切都是正常的。
这里的一切指：

nivida470和cuda11.4版本一致
cuda11.4和pytorch11.3版本一致【pytorch没有11.4，11.3可以适配】
环境变量设置正确
print(torch.__version__)显示 “XX+cu11.3”.

尝试：

卸载&重新安装cuda版本的pytorch
在环境变量中增加

export CUDA_VISIBLE_DEVICES=0
#也尝试了
export CUDA_VISIBLE_DEVICES=0,1,2,

更新nvidia-modprobe

apt-get install nvidia-modprobe

但都不能解决问题。

最终解决方案：

直到！！！我重启，重启大法好！

给print(torch.cuda.is_available()) False的兄弟们提供一个可能的解决方案