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
留言
張貼留言