Generally, the package works fine if you have installed latest nvidia
drivers and python-pytorch
. In my case, I installed python-pytorch-opt-cuda
and no error was reported.
However, if you encountered something like fatal error: cuda_runtime.h: No such file or directory
. Please check your environment variables (~/.bashrc) related to CUDA. I would recommend you directly install drivers from official Archlinux repositories, which is very convenient and avoids issues.
Pinned Comments
fanyujun commented on 2023-10-12 07:16 (UTC)
Generally, the package works fine if you have installed latest
nvidia
drivers andpython-pytorch
. In my case, I installedpython-pytorch-opt-cuda
and no error was reported.However, if you encountered something like
fatal error: cuda_runtime.h: No such file or directory
. Please check your environment variables (~/.bashrc) related to CUDA. I would recommend you directly install drivers from official Archlinux repositories, which is very convenient and avoids issues.