我在一台装有Nvidia GTX 1050 GPU的Windows 10机器上安装了PyTorch。我已经安装了CUDA工具包,并使用Nvidia instructions进行了测试,一切顺利,包括执行建议的测试。
但是,torch.cuda.is_available()
返回False
。如何修复此问题?
我在一台装有Nvidia GTX 1050 GPU的Windows 10机器上安装了PyTorch。我已经安装了CUDA工具包,并使用Nvidia instructions进行了测试,一切顺利,包括执行建议的测试。
但是,torch.cuda.is_available()
返回False
。如何修复此问题?
4条答案
按热度按时间yzckvree1#
我也遇到了同样的问题,运行这个=〉
a=torch.cuda.FloatTensor()
,给出了Assert错误AssertionError: Torch not compiled with CUDA enabled
......这就澄清了我在没有cuda的情况下运行pytorch。步骤:
1.通过调用以下命令,确保您已经卸载了Pytorch:
pip uninstall torch
1.转到https://pytorch.org/get-started/locally/并选择系统配置(如图所示)。
1.从
Run this command
对话框中复制完全相同的命令,并在终端上运行它。ocebsuys2#
我遇到了同样的问题,结果是我通过运行https://pytorch.org/get-started/locally/提供的命令安装了一个仅CPU版本。
如果你像我一样安装了CUDA 10.2,网站可能会给予你
pip install torch===1.7.1 torchvision===0.8.2 torchaudio===0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
,没有明确指定CPU或GPU。在我的例子中,它将我路由到
cpu/torch-1.7.1%2Bcpu-cp38-cp38-win_amd64.whl
,这是一个CPU版本,而不是我的CUDA 10.2的cu102/torch-1.7.1-cp38-cp38-win_amd64.whl
。我的解决方法是,直接转到
https://download.pytorch.org/whl/torch_stable.html
并下载与您的环境匹配的二进制文件,然后只需从下载文件夹安装:pip install --no-cache-dir --force-reinstall torch===1.7.1 torchvision===0.8.2 -f .\Downloads\
3ks5zfa03#
torch.cuda.is_available()
导致False
的原因是pytorch
和cudatoolkit
的版本之间不兼容。截至2022年6月,当前版本的pytorch与cudatoolkit = 11.3兼容,而当前的cuda工具包版本= 11.7. Source
解决方案:
1.运行
conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
来安装pytorch。1.从https://developer.nvidia.com/cuda-11.3.0-download-archive安装CUDA 11.3版本。
你可以走了。
原始答案贴在这里:https://stackoverflow.com/a/72650265/10468354。以下内容仅供参考:
piah890a4#
如果其他解决方案似乎不起作用,则建议重新启动运行库。