nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2021 NVIDIA Corporation
Built on Sun_Feb_28_22:34:44_PST_2021
Cuda compilation tools, release 10.2, V10.2.300
Build cuda_10.2_r440.TC440_70.29663091_0
PyTorch for Jetson - version 1.10 now available
- PyTorch v1.7 - torchvision v0.8.1
sudo apt-cache show nvidia-jetpack
It will show something like:
Package: nvidia-jetpack
Version: 4.4-b144
Architecture: arm64
Maintainer: NVIDIA Corporation
-------------------------------------
cuda version
$ cat /usr/local/cuda/version.txt
CUDA Version 10.2.300
ls /usr/local/cuda*
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2-------------------------------------workable pytorch in TX2 w/ python3.6 CUDA 10.2, jetpack 4.4https://developer.nvidia.com/embedded/downloads----------------------------------
pip3 install --no-deps torchvision
Collecting torchvision
Using cached https://files.pythonhosted.org/packages/fb/01/03fd7e503c16b3dc262483e5555ad40974ab5da8b9879e164b56c1f4ef6f/torchvision-0.2.2.post3-py2.py3-none-any.whl
Installing collected packages: torchvision
Successfully installed torchvision-0.2.2.post3