setup CUDA path

First of all, check your CUDA:

nvcc -V or nvcc --version

If there is nothing:

export CUDA_HOME=/usr/local/cuda-<version>  # Replace <version> with your CUDA version
export PATH=$CUDA_HOME/bin:$PATH
export LD_LIBRARY_PATH=$CUDA_HOME/lib64:$LD_LIBRARY_PATH
posted @ 2024-02-28 21:30  kingchou007  阅读(3)  评论(0编辑  收藏  举报