pytorch安装Torch not compiled with CUDA enabled
时间: 2023-11-08 21:03:00 浏览: 491
解决AssertionError Torch not compiled with CUDA enabled.docx
这个问题可能是由于您安装的Torch没有启用CUDA支持所致。为了使用CUDA加速,您需要安装与您的GPU驱动程序相匹配的PyTorch版本。
首先,请确保您的计算机上已经正确安装了NVIDIA GPU驱动程序。然后按照以下步骤安装支持CUDA的PyTorch:
1. 通过 pip 安装 torch:
```
pip install torch==1.7.0
```
2. 如果您使用的是 CUDA 10.1,请运行以下命令:
```
pip install torch==1.7.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html
```
3. 如果您使用的是其他 CUDA 版本,请根据您的 CUDA 版本选择相应的命令行。例如,如果您使用的是 CUDA 10.2,请运行以下命令:
```
pip install torch==1.7.0+cu102 -f https://download.pytorch.org/whl/torch_stable.html
```
请注意,上述命令中的版本号"1.7.0"是示例版本号,您可以根据您的实际需求选择特定的版本号。
阅读全文