[일반] 혹시 ARM64 aubuntu 리눅스 쪽 Cuda cudnn 설치하신 분

[익명(180.233)]

arch : aarch64(ARM64)

GPU &CUDA : 

nvcc -V

cudnn : cudnn-local-repo-ubuntu2204-9.2.0

https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=arm64-sbsa&Compilation=Native&Distribution=Ubuntu&target_version=22.04&target_type=deb_local

cuDNN 9.2.0 Downloads

torch 

그 후 아래 python 들어가 서 실행해보면 False 나옵니다..

import torch torch.cuda.is_available()

무슨 문제인지 아시는분?

[익명(222.113)]

torch 2.3.1은 cuda 11.8 or 12.1만 지원해서 그런듯 cuda version을 낮춰야 할 것 같음