Debian9.8 安裝 pytorch

Debian9.8 安裝 pytorch (不安裝cuda):
apt install -y git python3-dev
git clone --recursive https://github.com/pytorch/pytorch

export NO_CUDA=1

export USE_NUMPY=1

export USE_FFMPEG=1

export USE_OPENCV=1

export USE_REDIS=1
cd pytorch

pip install opencv-python numpy scipy tensorflow scikit-learn matplotlib redis ffmpeg

pip install pyyaml

pip install -r requirements.txt
python3 setup.py install

pip install torchvision

 

或者直接按照pytorch官網上https://pytorch.org/選擇你要安裝的，如：

pip install https://download.pytorch.org/whl/cpu/torch-1.0.1-cp35-cp35m-win_amd64.whl

 

用命令查看pytorch適合安裝成功：

pip list | grep torch

 

簡單例子：

import torch

x = torch.empty(5, 3)
print(x)
 

輸出：

哈哈，之後就能夠用pytorch開發了