Mac 安装pytorch

 

conda deactivate py3.6

 

 

1. 类似于安装 tensorflow，先创建一个名字叫pytorch3.6的虚拟环境

conda create -n py3.6 python=3.6

 

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

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

conda install pytorch torchvision -c pytorch

 

 

pip install scipy

 

这样安装的是pytorch 1.4.0

如何查看当前pytorch版本:

import torch
print(torch.__version__)
# 1.4.0

 

但是我git下的项目所需要的pytorch是0.4.0或0.4.1

The model is built in PyTorch 0.4.0, PyTorch 0.4.1 and tested on Ubuntu 14.04/16.04 environment (Python3.6, CUDA9.0, cuDNN5.1)

卸载掉pytorch：

conda uninstall pytorch

安装：

conda install pytorch=0.4.1

 

安装一些其他库，比如：torchvision

基本上都是 pip install torchvision

 

查看pytorch版本号：

import torch
print(torch.__version__)  #注意是双下划线