Anaconda install pytorch, caffe2 和 onnx

Ubuntu 安裝 cuda, pytorch, caffe2 和 onnx 遇上很多問題,後來發現 Anaconda 可以輕鬆安裝帶 gpu 的版本真是太方便了。

1. Install Anaconda:

$ curl -O https://repo.anaconda.com/archive/Anaconda3-5.2.0-Linux-x86_64.sh
$ sh Anaconda3-5.2.0-Linux-x86_64.sh

重新開啟終端機,查看 python 或 python3:

$ python

Python 3.6.5 |Anaconda, Inc.| (default, Apr 29 2018, 16:14:56) 
[GCC 7.2.0] on linux
Type "help", "copyright", "credits" or "license" for more information.


2. Install pytorch with cuda9.0:

$ conda install pytorch torchvision cuda90 -c pytorch

3. Install caffe2 with cuda9.0:

$ conda install -c caffe2 caffe2-cuda9.0-cudnn7

4. Install onnx:

$ conda install -c conda-forge onnx

Reference:

[1] https://pytorch.org/get-started/locally/

[2] https://caffe2.ai/docs/getting-started.html?platform=ubuntu&configuration=prebuilt

[3] https://github.com/onnx/onnx

留言

熱門文章