安装GPUconda install pytorch torchvision torchaudio -c pytorch
时间: 2023-07-03 19:31:12 浏览: 161
Anaconda下pytorch环境的创建,pytorch,torchvision的安装(cpu)
5星 · 资源好评率100%
这个命令可以安装PyTorch、torchvision和torchaudio,其中-c pytorch表示从pytorch官方渠道安装。如果你想安装GPU版本的PyTorch,可以使用以下命令:
```
conda install pytorch torchvision torchaudio cudatoolkit=<version> -c pytorch
```
其中`<version>`是你的CUDA版本号,比如如果你的CUDA版本是10.2,那么命令就应该是:
```
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch
```
安装完毕后,你可以通过以下命令来验证是否安装成功:
```
import torch
print(torch.cuda.is_available())
```
如果输出True,则表示你已经成功安装了GPU版本的PyTorch。
阅读全文