cuda10.2对应的pytorch gpu
时间: 2023-10-21 12:04:45 浏览: 225
1.10.0GPU版本pytorch环境,conda安装支持cuda10.2和11.3 pip安装支持cuda10.2和11.1
对于 CUDA 10.2,可以使用 PyTorch 的 GPU 版本。你可以通过以下命令安装适用于 CUDA 10.2 的 PyTorch GPU 版本:
```shell
pip install torch==1.7.1+cu102 torchvision==0.8.2+cu102 torchaudio==0.7.2 -f https://download.pytorch.org/whl/torch_stable.html
```
请注意,上述命令中的版本号可能会因为时间的推移而有所更改,因此请确保查看 PyTorch 官方网站以获取最新版本和安装说明。
阅读全文