我在Windows 10中安装了GPU和CUDA,但是Pytorch的torch.cuda.is_available()返回false;我该如何纠正这个错误?

fkaflof6  于 2023-01-13  发布在  Windows
关注(0)|答案(4)|浏览(700)

我在一台装有Nvidia GTX 1050 GPU的Windows 10机器上安装了PyTorch。我已经安装了CUDA工具包,并使用Nvidia instructions进行了测试,一切顺利,包括执行建议的测试。
但是,torch.cuda.is_available()返回False。如何修复此问题?

yzckvree

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对话框中复制完全相同的命令,并在终端上运行它。

ocebsuys

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\

3ks5zfa0

3ks5zfa03#

torch.cuda.is_available()导致False的原因是pytorchcudatoolkit的版本之间不兼容。
截至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。以下内容仅供参考:

piah890a

piah890a4#

如果其他解决方案似乎不起作用,则建议重新启动运行库。

相关问题