開発のヒホ

iOSとかAndroidとかのアプリを開発するのに四苦八苦するブログ

素直なCUDA環境でGPUを使用可能にする環境変数メモ

 色んなサーバでGPU環境を整えていると、そのたびに調べないといけないのでメモ。もし素直じゃないCUDA環境でも、CUDA_ROOTを変えたりlib64を変更するだけでなんとかなるはず。これを書いた.bash_profileを毎回読みこめばCUDA環境が整うはず。

export CUDA_ROOT=/usr/local/cuda
export LD_LIBRARY_PATH=${CUDA_ROOT}/lib64
export CUDA_INC_DIR=/usr/local/cuda/include
export CUDA_ENABLE_GL=False # ?
export CUDA_ENABLE_CURAND=True # ?
export CUDADRV_LIB_DIR=${CUDA_ROOT}/lib64
export CUDADRV_LIBNAME=cuda
export CUDART_LIB_DIR=${CUDA_ROOT}/lib64
export PATH=/usr/local/cuda/bin:$PATH