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 16:22:33 浏览: 357
机器学习,pytorch英伟达显卡驱动,cndann,适用于cuda12.1以上
可以使用以下语句安装对应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`通道。
阅读全文