cuda11.1适合什么版本的pytorch?有没有在conda中的安装命令
时间: 2024-04-08 19:31:16 浏览: 107
win10+cuda10.0+pytorch安装
5星 · 资源好评率100%
CUDA 11.1 目前适用于 PyTorch 1.7.0 及更高版本。你可以使用以下命令在 conda 中安装适用于 CUDA 11.1 的 PyTorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch
```
这将安装适用于 CUDA 11.1 的最新版本的 PyTorch。
阅读全文