素直な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