pip安装pytorch cuda10 2
时间: 2023-11-16 18:47:58 浏览: 105
win10+cuda10.0+pytorch安装
5星 · 资源好评率100%
你可以通过以下命令来安装 PyTorch 在 CUDA 10.2 上运行:
```
pip install torch torchvision torchaudio -f https://download.pytorch.org/whl/cu102/torch_stable.html
```
请注意,这需要已经安装配置好 CUDA 10.2 的 GPU 环境。如果你没有安装 CUDA,或者你的 CUDA 版本不是 10.2,请先安装适合你的 CUDA 版本,然后使用相应的 PyTorch 版本。
阅读全文