PyTorch中torch.cuda.is_available()返回False,CUDA版本匹配问题求助
PyTorch CUDA配置问题求助
我在为PyTorch开发配置CUDA时遇到问题,急需帮助。我拥有NVIDIA显卡,使用Python 3.8,为安装带正确CUDA集成的PyTorch,执行了命令:
conda install pytorch torchvision cudatoolkit=10.1 -c python
但torch.cuda.is_available()始终返回False。
以下是我的版本信息:
- 执行
nvcc --version的输出:
nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2019 NVIDIA Corporation Built on Sun_Jul_28_19:07:16_PDT_2019 Cuda compilation tools, release 10.1, V10.1.243
- 执行
nvidia-smi的输出:
NVIDIA-SMI 510.47.03 Driver Version: 510.47.03 CUDA Version: 11.6
恳请各位帮忙解决此问题。
内容的提问来源于stack exchange,提问作者Krokus
相关产品推荐
相关产品推荐

