Skip to content
Branch: master
Find file Copy path
Find file Copy path
1 contributor

Users who have contributed to this file

87 lines (73 sloc) 2.68 KB
sudo apt install libopenblas-dev libatlas-dev liblapack-dev
sudo apt install liblapacke-dev checkinstall # For OpenCV
sudo apt-get install python-pip
pip install --upgrade pip==9.0.1
sudo apt-get install python-dev
sudo pip install numpy==1.15.1
sudo pip install scipy # ~20-30 min
sudo apt-get install python-matplotlib
sudo pip install matplotlib==2.2.3
sudo pip install pyyaml
sudo pip install scikit-build
sudo apt-get -y install cmake
sudo apt install libffi-dev
sudo pip install cffi
#add couple of lines to the end of ~/.bashrc or ~/.profile
sudo gedit ~/.bashrc
export CUDNN_LIB_DIR=/usr/lib/aarch64-linux-gnu
export CUDNN_INCLUDE_DIR=/usr/include
source ~/.bashrc
sudo add-apt-repository universe
sudo apt-get update
sudo apt-get install alien
sudo apt-get install nano
sudo alien ninja-1.8.2-3.mga7.aarch64.rpm
#If previous line fails, proceed to <$sudo dpkg -i ninja-1.8.2-3.mga7.aarch64.deb>
sudo alien -g ninja-1.8.2-3.mga7.aarch64.rpm
cd ninja-1.8.2
sudo nano debian/control
#at architecture, add arm64 after aarch64 (aarch64, arm64)
sudo debian/rules binary
cd ..
sudo dpkg -i ninja_1.8.2-4_arm64.deb
sudo apt install ninja-build
cd ..
git clone
cd pytorch
sudo pip install -U setuptools
sudo pip install -r requirements.txt
git submodule update --init --recursive
sudo apt-get install libfreetype6-dev pkg-config libpng12-dev
python build_deps
sudo gedit /pytorch/CMakeList.txt
> CmakeLists.txt : Change NCCL to 'Off' on line 97
sudo gedit /pytorch/
> Add USE_NCCL = False below line 198
sudo gedit /pytorch/tools/setup_helpers/
> : Change USE_SYSTEM_NCCL to 'False' on line 8
Change NCCL to 'False' on line 78
sudo gedit /pytorch/torch/csrc/cuda/nccl.h
> nccl.h : Comment self-include on line 8
Comment entire code from line 21 to 28
sudo gedit torch/csrc/distributed/c10d/ddp.cpp
> ddp.cpp : Comment nccl.h include on line 6
Comment torch::cuda::nccl::reduce on line 148
sudo nvpmodel -m 0
sudo DEBUG=1 python build develop
#If build fails before 100%
sudo python clean
sudo DEBUG=1 python build develop
#elif build fails after 100%
sudo python clean
sudo DEBUG=1 python develop
sudo apt clean
sudo apt-get install libjpeg-dev zlib1g-dev
git clone
cd Pillow/
sudo python install
sudo pip install pandas # ~20-30 min
sudo pip install Cython
sudo pip install scikit-image
sudo apt install python-sklearn
sudo pip --no-cache-dir install torchvision
You can’t perform that action at this time.