Hi,
I have installed cuda, but get that cuda is not available
print(torch.cuda.is_available())
False
How can i fix this?
installed torch by following this guide; Installing PyTorch for Jetson Platform - NVIDIA Docs
Hi,
I have installed cuda, but get that cuda is not available
print(torch.cuda.is_available())
False
How can i fix this?
installed torch by following this guide; Installing PyTorch for Jetson Platform - NVIDIA Docs
Hi,
Could you share which JetPack you use?
$ apt show nvidia-jetpack
And the steps that you used for installing PyTorch?
Thanks.
apt show nvidia-jetpack
Package: nvidia-jetpack
Version: 6.2+b77
Priority: standard
Section: metapackages
Source: nvidia-jetpack (6.2)
Maintainer: NVIDIA Corporation
ah, it seems like pytorch was not correctly installed after all. I installed cusparsel, and do the following:
export TORCH_INSTALL=https://developer.download.nvidia.cn/compute/redist/jp/v62/pytorch/torch-2.0.0+nv23.05-cp38-cp38-linux_aarch64.whl
python3 -m pip install --upgrade pip; python3 -m pip install numpy==’1.26.1’; python3 -m pip install --no-cache $TORCH_INSTALL
what solved this issue for us was to run this:
python -m pip uninstall torch
python -m pip cache purge
# Use 11.7, it should be compatible
python -m pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117
This topic was automatically closed 14 days after the last reply. New replies are no longer allowed.