Cuda Compatible torchvision for PyTorch v2.2.0

Robotics & Edge Computing Jetson & Embedded Systems Jetson Orin Nano
,
1

I installed PyTorch v2.2.0

wget https://developer.download.nvidia.cn/compute/redist/jp/v60dp/pytorch/torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl -O torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl
sudo apt-get install python3-pip libopenblas-base libopenmpi-dev libomp-dev
pip3 install Cython==0.29.36
pip3 install numpy torch-2.2.0a0+6a974be.nv23.11-cp310-cp310-linux_aarch64.whl



sudo apt-get install libjpeg-dev zlib1g-dev libpython3-dev libopenblas-dev libavcodec-dev libavformat-dev libswscale-dev
git clone --branch v0.17 https://github.com/pytorch/vision torchvision   # see below for version of torchvision to download
cd torchvision
export BUILD_VERSION=0.17.0  # where 0.x.0 is the torchvision version  
python3 setup.py install --user
cd ../  # attempting to load torchvision from build dir will result in import error
pip install 'pillow<7' # always needed for Python 2.7, not needed torchvision v0.5.0+ with Python 3.6

Pytorch installation was successful but torchvision was unsuccessful

fatal: Remote branch v0.17 not found in the upstream orgin

what should i do install torchvision copatible to pytorch v2.2.0

3

Hi,

There is no v0.17 TorchVision branch available yet.
Maybe you can give rc version a try.

Thanks.

5

This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.