PyTorch Mac 安装教程

打开 pytorch.org 官网,可以选择安装器,Python版本,CUDA 版本。

例如 Mac Conda Python3.6 CUDA 9.0 的安装命令为:

conda install pytorch torchvision -c pytorch

如果是更加常见的 pip 安装的命令则是:

pip3 install torch torchvision

发表评论

电子邮件地址不会被公开。 必填项已用*标注