conda安装pytorch
时间: 2023-06-29 14:20:03 浏览: 119
使用anaconda安装pytorch的实现步骤
你可以使用以下命令在conda环境中安装pytorch:
```
conda install pytorch torchvision torchaudio cudatoolkit=<version>
```
其中`<version>`是你的CUDA版本号,如`10.2`或`11.1`等。
如果你使用的是CPU版本,可以使用以下命令:
```
conda install pytorch-cpu torchvision-cpu torchaudio-cpu -c pytorch
```
注意,这里使用了`-c pytorch`选项来指定从pytorch的conda channel安装。
阅读全文