pytorch notes

master
server 2020-02-11 09:25:36 -07:00
parent 3f3f8d3af2
commit 0d25371eac
1 changed files with 39 additions and 0 deletions

View File

@ -141,6 +141,17 @@ CAFFE2_USE_MSVC_STATIC_RUNTIME OFF ??
BUILD_CAFFE2_MOBILE OFF
```
# More misc...
```
git checkout v1.4.0
apt install python3-dev python3-numpy python-numpy
# uh
apt install libcaffe-cpu-dev
```
Seems to be using not python3??
# Misc
```
[E init_intrinsics_check.cc:43] CPU feature avx is present on your machine, but the Caffe2 binary is not compiled with it. It means you may not get the full speed of your CPU.
@ -148,6 +159,34 @@ BUILD_CAFFE2_MOBILE OFF
# Build PyTorch `pip`
Lets get old Python 2 out of here:
```
apt autoremove --purge python2 python2-minimal python2.7 python2.7-minimal libpython2.7 libpython2.7-minimal libpython2.7-stdlib
git checkout v1.4.0
apt install python3-pip python3-setuptools
# Docs recommend this, but mkl is proprietary:
pip3 install --user numpy ninja pyyaml mkl mkl-include setuptools cmake cffi
# Try:
pip3 install --user --upgrade pip
# Set that new pip in your path, ~/.local/bin/ in ~/.bashrc:
export PATH="~/.local/bin:/usr/lib/ccache:$PATH"
pip3 install --user cmake
pip3 install --user numpy ninja pyyaml setuptools cmake cffi
# From upstream docs:
pip install torch==1.4.0+cpu torchvision==0.5.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
```
# Build with Python Setup
```
git submodule sync
git submodule update --init --recursive
python3 setup.py build --cmake-only
ccmake build
python3 setup.py install --user
```
# Proprietary Bits
Parts to avoid in the PyTorch ecosystem: