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__)  #注意是雙下划線