TensorFlowには、cudaとCuDNNをインストールしたいと思います。 Ubuntu 16.04でそれを行うにはどうすればよいですか?
ベストアンサー
ステップ0:標準リポジトリからcudaをインストールします。 (Ubuntu 16.04にCUDAをインストールするにはどうすればよいですか?)
ステップ1:nvidia開発者アカウントを登録し、ここからcudnnをダウンロードします(約80 MB)
ステップ2:cudaのインストール場所を確認します。リポジトリからのインストールの場合は、/usr/lib/...
です。 および/usr/include
。それ以外の場合は、/usr/local/cuda/
になります または/usr/local/cuda-<version>
。 which nvcc
で確認できます またはldconfig -p | grep cuda
ステップ3:ファイルをコピーする:
リポジトリのインストール:
$ cd folder/extracted/contents
$ sudo cp -P include/cudnn.h /usr/include
$ sudo cp -P lib64/libcudnn* /usr/lib/x86_64-linux-gnu/
$ sudo chmod a+r /usr/lib/x86_64-linux-gnu/libcudnn*
Runfileのインストール:
$ cd folder/extracted/contents
$ sudo cp include/cudnn.h /usr/local/cuda/include
$ sudo cp lib64/libcudnn* /usr/local/cuda/lib64
$ sudo chmod a+r /usr/local/cuda/lib64/libcudnn*