CUDA Version: 11.2 pytorch安装指令
时间: 2023-09-20 20:07:15 浏览: 526
cuda cudann pytorch安装.doc
您可以使用以下指令安装适用于 CUDA 11.2 的 PyTorch:
```python
pip install torch==1.8.0+cu112 torchvision==0.9.0+cu112 torchaudio==0.8.0 -f https://download.pytorch.org/whl/cu112/torch_stable.html
```
这将安装 PyTorch 1.8.0 版本,以及相应的 torchvision 和 torchaudio。请确保您的 CUDA 版本和指定的 cu112 版本匹配。
阅读全文