conda install pytorch==2.0.0 torchvision==0.15.0 torchaudio==2.0.0 pytorch-cuda=11.8 -c pytorch -c nvidia是对应CUDA11.8的,请查找对应CUDA12.0的语句
时间: 2023-09-15 15:22:33 浏览: 314
CUDA10.1向下兼容10.0,完美运行官方TensorFlow、Pytorch_pytorchcuda10.0,cuda10.0可以升级
5星 · 资源好评率100%
可以使用以下语句安装对应CUDA12.0的PyTorch和相关库:
```
conda install pytorch==1.9.0 torchvision==0.10.0 torchaudio==0.9.0 cudatoolkit=12.0 -c pytorch
```
这条语句会安装PyTorch 1.9.0和对应的torchvision和torchaudio,同时指定使用CUDA 12.0。注意,此语句中不需要指定`pytorch-cuda`和`nvidia`通道。
阅读全文