Jetson TX2 Pytorch 1.1 install

sudo apt-get install cmake python3-pip
cd
git clone -b v1.1.0 https://github.com/pytorch/pytorch
git submodule update --init --recursive
cd pytorch
sudo pip3 install -r requirements.txt
vim tools/setup_helpers/nccl.py
USE_NCCL = False
sudo python3 setup.py install

설치를 조금 더 빠르게 하기 위해 설치전에 아래 코드를 실행한 후 설치

sudo nvpmodel -m 0
sudo jetson_clocks.sh

댓글을 남겨주세요~