CUDA not available on aarch64

just to verify. the aarch64 release are not compiled with cuda enabled? Is this true?

The installation:
python3 -m pip install torch --extra-index-url

The output:

root@localhost:~# python3 -m torch.utils.collect_env
Collecting environment information...
PyTorch version: 1.10.2
Is debug build: False
CUDA used to build PyTorch: None
ROCM used to build PyTorch: N/A

OS: Ubuntu 18.04.5 LTS (aarch64)
GCC version: Could not collect
Clang version: Could not collect
CMake version: Could not collect
Libc version: glibc-2.25

Python version: 3.6.9 (default, Mar 15 2022, 13:55:28)  [GCC 8.4.0] (64-bit runtime)
Python platform: Linux-4.9.140-aarch64-with-Ubuntu-18.04-bionic
Is CUDA available: False
CUDA runtime version: No CUDA
GPU models and configuration: No CUDA
Nvidia driver version: No CUDA
cuDNN version: No CUDA
HIP runtime version: N/A
MIOpen runtime version: N/A

Versions of relevant libraries:
[pip3] numpy==1.19.5
[pip3] torch==1.10.2
[conda] Could not collect

Yes, I think that’s the case as the aarch64 builds might have been created for Mac.
You could use the NGC containers which provide ARM support.

1 Like