torch.cuda.device_count()返回0
torch.cuda.device_count()返回0
提问于 2019-03-24 12:45:36
我已经通过以下途径安装了虚拟环境下的py手电筒：
pip3 install https://download.pytorch.org/whl/cu100/torch-1.0.1.post2-cp35-cp35m-linux_x86_64.whl
pip3 install torchvision但出于某种原因：
import torch
torch.cuda.device_count()
0
torch.cuda.is_available()
False似乎是因为某种原因发现了cuda 9：
torch._C._cuda_getDriverVersion()
9010我的系统：
lsb_release -a
No LSB modules are available.
Distributor ID: Ubuntu
Description: Ubuntu 16.04.5 LTS
Release: 16.04
Codename: xenial
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2019 NVIDIA Corporation
Built on Fri_Feb__8_19:08:17_PST_2019
Cuda compilation tools, release 10.1, V10.1.105
Python 3.5.2 (default, Nov 23 2017, 16:37:01)
[GCC 5.4.0 20160609] on linux
ls -l /usr/local/ | grep cuda
lrwxrwxrwx 1 root root 21 Mar 24 02:05 cuda -> /usr/local/cuda-10.0/
drwxr-xr-x 17 root root 4096 Mar 24 02:06 cuda-10.0
drwxr-xr-x 15 root root 4096 Nov 29 23:03 cuda-9.0
drwxr-xr-x 15 root root 4096 Jan 10 2018 cuda-9.1
nvidia-smi -q | grep "Product Name"
Product Name : GeForce GTX TITAN X
Product Name : GeForce GTX 1080 Ti回答 1
Stack Overflow用户
发布于 2022-06-30 07:05:54
如果torch._C._cuda_getDriverVersion()给出了驱动程序值，请尝试使用torch._C._cuda_getDeviceCount()来理解更多内容。
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接：
https://stackoverflow.com/questions/55323930
复制相关文章
相似问题
腾讯云开发者
