intelligent-machine-learning / intelligent-machine-learning/atorch

使用registry.cn-hangzhou.aliyuncs.com/atorch/atorch-open-20240430:pt210构建的容器,出现cuda error

Open
#2 1 comment 0 reactions 0 assignees View on GitHub
Dominant language
Python
Stars
62
Forks
6
PR merge metrics
No merged PRs in 30d

Description

调用torch.cuda.is_available()报错如下
/opt/conda/lib/python3.8/site-packages/torch/cuda/__init__.py:138: UserWarning: CUDA initialization: Unexpected error from cudaGetDeviceCount(). Did you run some cuda functions before calling NumCudaDevices() that might have already set an error? Error 803: system has unsupported display driver / cuda driver combination (Triggered internally at ../c10/cuda/CUDAFunctions.cpp:108.)
return torch._C._cuda_getDeviceCount() > 0

Contributor guide

No contributing guide indexed for this repository

Assessment

This issue has not been assessed yet.

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.