鲲鹏论坛

在鲲鹏920 和安装银河麒麟V10 SP3 的服务器上搭建PyTorch环境遇到一些问题, nvidia-driver、cuda和cudnn成功安装后，torch.cuda.is_available() 返回 false:

Python 3.7.12 | packaged by conda-forge | (default, Oct 26 2021, 06:22:24)
[GCC 9.4.0] on linux
Type "help", "copyright", "credits" or "license" for more information.
>>> import torch
>>> torch.cuda.is_available()
False

torch 版本如下:

# pip list|grep torch
torch              1.13.1

显卡驱动和cuda均使用 cuda_11.2.2_460.32.03_linux_sbsa.run 安装.

nvidia-smi返回内容如下:

+-----------------------------------------------------------------------------+
| NVIDIA-SMI 460.32.03    Driver Version: 460.32.03    CUDA Version: 11.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  Tesla T4            Off  | 00000000:01:00.0 Off |                    0 |
| N/A   39C    P8     9W /  70W |      4MiB / 15109MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+
|   1  Tesla T4            Off  | 00000000:02:00.0 Off |                    0 |
| N/A   39C    P8     9W /  70W |      4MiB / 15109MiB |      0%      Default |
|                               |                      |                  N/A |
+-------------------------------+----------------------+----------------------+

+-----------------------------------------------------------------------------+
| Processes:                                                                  |
|  GPU   GI   CI        PID   Type   Process name                  GPU Memory |
|        ID   ID                                                   Usage      |
|=============================================================================|
|    0   N/A  N/A      2848      G   /usr/libexec/Xorg                   4MiB |
|    1   N/A  N/A      2848      G   /usr/libexec/Xorg                   4MiB |
+-----------------------------------------------------------------------------+

nvcc -V 返回内容如下:

nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_14_21:13:57_PST_2021
Cuda compilation tools, release 11.2, V11.2.152
Build cuda_11.2.r11.2/compiler.29618528_0

不明白为啥torch.cuda.is_available() 返回 false, 跑demo脚本打印出的信息也提示使用 CPU

关于鲲鹏

新闻与活动

交流与资讯

支持与服务

开源社区