cuda12.1安装什么torch版本
时间: 2023-09-29 18:02:12 浏览: 202
机器学习,pytorch英伟达显卡驱动,cndann,适用于cuda12.1以上
对于 CUDA 12.1,您可以安装适用于该版本的 PyTorch 1.10.0 版本。您可以使用以下命令安装:
```bash
pip install torch==1.10.0+cu112 -f https://download.pytorch.org/whl/torch_stable.html
```
请注意,此命令需要具有适
阅读全文