conda安装pytorch、torchvision的命令,安装GPU版本。
时间: 2024-05-15 10:17:43 浏览: 60
Anaconda下pytorch环境的创建,pytorch,torchvision的安装(cpu)
5星 · 资源好评率100%
您可以使用以下命令安装conda版本的pytorch和torchvision GPU版本:
```
conda install pytorch torchvision cudatoolkit=10.2 -c pytorch
```
请注意,这个命令假设您的Cuda版本是 10.2。如果不是,请对“cudatoolkit”参数进行相应的更改。
阅读全文