Installing CUDA9 + CuDNN7 + Torch on CentOS 7 (Japanese post + English post)
このポストの目的
下記の自著記事を基にして、CUDA8 + CuDNN5.1はinstallした計算機があります。
今度はこのマシンのCUDA, CuDNNを更新します。
Purpose of this post:
update the CUDA (8 --> 9) and CuDNN (5 -->7) on CentOS 7.
作業開始!
CUDA, CuDNNのダウンロード
まずはnvidiaの公式ページよりCUDAのrpm(network install)とCuDNNのtgzを確保します。
Download the .rpm file of CUDA9 (network install) and .tgz file of CuDNN7.
CUDA Toolkit Download | NVIDIA Developer
NVIDIA cuDNN | NVIDIA Developer
CUDA、CuDNNのインストール
特に何も考えずにrpmインストールです。
Simplu install CUDA from RPM.
sudo rpm --install cuda-repo-rhel7-9.0.176-1.x86_64.rpm
sudo yum clean expire-cache
sudo yum install cuda
環境変数をアップデートします。sourceを忘れずに。
Update the environmental variables. Do not forget source-ing!
export PATH="/usr/local/cuda-9.0/bin:$PATH"
# THIS IS OBSOLETE export PATH="/usr/local/cuda-8.0/bin:$PATH"
export LD_LIBRARY_PATH="/usr/local/cuda-9.0/lib64:$LD_LIBRARY_PATH"# THIS IS OBSOLETE export LD_LIBRARY_PATH="/usr/local/cuda-8.0/lib64:$LD_LIBRARY_PATH"
CuDNNについては、.tgzファイルを解凍して、出来上がるcudaディレクトリの中身を/usr/local/cudaディレクトリにコピーするだけです。
Copy the contents of the CuDNN .tgz file to /usr/local/cuda.
sudo cp -a cuda/lib64/* /usr/local/cuda/lib64
sudo cp -a cuda/include/cudnn.h /usr/local/cuda/include
TorchのRe-install
事前情報を得ていたので、以下の環境変数を設定。
Set the following varaible.
export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
Torchディレクトリからinstall. 躓いたら、./clean.shをしてからもう一度やってみましょう。
Install. If you failed, try ./clean.sh, then update all relevant packages and libraries.
cd ~/torch
./install-deps
./update.sh
./clean.sh
export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"
./install.sh
<<After successful installation, do luarocks for torch packages>>