Ubuntu系统GTX1070移动版torch.cuda.is_available()返回False求助

系统信息 #

• GPU：GTX 1070 Max-Q
• 操作系统：Ubuntu 22.04.3 LTS
• 显卡驱动：535.129.03
• Python版本：3.11.5
• nvidia-smi输出：

+---------------------------------------------------------------------------------------+
| NVIDIA-SMI 535.129.03             Driver Version: 535.129.03   CUDA Version: 12.2     |
|-----------------------------------------+----------------------+----------------------+
| GPU  Name                 Persistence-M | Bus-Id        Disp.A | Volatile Uncorr. ECC |
| Fan  Temp   Perf          Pwr:Usage/Cap |         Memory-Usage | GPU-Util  Compute M. |
|                                         |                      |               MIG M. |
|=========================================+======================+======================|
|   0  NVIDIA GeForce GTX 1070 ...    Off | 00000000:01:00.0 Off |                  N/A |
| N/A   39C    P8               8W /  90W |      6MiB /  8192MiB |      0%      Default |
|                                         |                      |                  N/A |
+-----------------------------------------+----------------------+----------------------+
                                                                                         
+---------------------------------------------------------------------------------------+
| Processes:                                                                            |
|  GPU   GI   CI        PID   Type   Process name                            GPU Memory |
|        ID   ID                                                             Usage      |
|=======================================================================================|
|    0   N/A  N/A      1979      G   /usr/lib/xorg/Xorg                            4MiB |
+---------------------------------------------------------------------------------------+

已尝试的操作（每次均通过conda create --name env_name python=3.11创建全新环境） #

• conda install pytorch cudatoolkit -c pytorch
• conda install pytorch torchvision torchaudio cudatoolkit=11.8 -c pytorch -c nvidia
• pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
• pip3 install torch torchvision torchaudio

问题描述 #

在Anaconda环境中安装PyTorch及对应cudatoolkit后，在Jupyter Notebook单元格中运行torch.cuda.is_available()始终返回False。无法确定问题根源是版本不兼容、GPU支持的CUDA版本限制还是PyTorch版本适配问题。根据NVIDIA CUDA发布说明及nvidia-smi输出，显卡驱动支持CUDA 12.2，理论上可兼容CUDA 11.8或12.1版本的PyTorch，但问题仍未解决，恳请提供可行解决方案。

内容的提问来源于stack exchange，提问作者Álvaro Lopez