anaconda安装gpu版本pytorch
时间: 2023-04-30 15:00:06 浏览: 223
Anaconda下pytorch环境的创建,pytorch,torchvision的安装(cpu)
5星 · 资源好评率100%
如果要在Anaconda中安装GPU版本的PyTorch,首先需要确保安装了CUDA和cuDNN。然后在Anaconda的命令行窗口中输入以下命令:
```
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
这将安装PyTorch和Torchvision,并指定使用CUDA 10.2。
阅读全文