pip安装pytorch cuda11.4版本的命令
时间: 2024-06-09 18:10:48 浏览: 450
torch.cuda.is_available()返回False的问题解决
你可以使用以下命令来安装 PyTorch 1.9 版本的 CUDA 11.4 版本:
```
pip install torch==1.9.0+cu111 -f https://download.pytorch.org/whl/cu111/torch_stable.html
```
请注意,这将在 CUDA 11.4 环境中安装 PyTorch 1.9。如果你的 CUDA 版本不同,你需要修改命令中的 `cu111` 参数为你所需的 CUDA 版本。
此外,你需要先安装适用于你的系统的 CUDA 和 cuDNN 版本,才能在你的系统上使用 PyTorch。请确保你已经正确安装了这些依赖项。
阅读全文