cuda cudnntensorflow-gpuインストールチュートリアル

1:まず、使用する必要のあるtensorflow-gpuバージョンに従って、インストールする必要のあるcudaバージョンとcudnnバージョンを決定します。

2:1.6.0 gpuをインストールする必要がある場合は、最初にcuda9.0とcudnn7.0をインストールする必要があります

3:cudaダウンロードアドレス:https://developer.nvidia.com/cuda-90-download-archive

sudo chmod +x cuda_9.0.176_384.81_linux.run # 为 cuda_9.0.176_384.81_linux.run 添加可执行权限
./cuda_9.0.176_384.81_linux.run # 安装 cuda_9.0.176_384.81_linux.run

/ etc / profileファイルを変更します

sudo vim /etc/profile

最後に追加

export PATH=/usr/local/cuda/bin:$PATH
export LD_LIBRARY_PATH=/usr/local/cuda/lib64$LD_LIBRARY_PATH
export CUDA_HOME=/usr/local/cuda

もう一度調達するだけです

source /etc/profile
sudo ldconfig

nvcc--versionはcudaのバージョンを表示できます

4:cudnnダウンロードアドレス:https://developer.nvidia.com/rdp/cudnn-download

sudo cp cuda/include/cudnn.h /usr/local/cuda-9.0/include
sudo cp cuda/lib64/libcudnn* /usr/local/cuda-9.0/lib64
sudo chmod a+r /usr/local/cuda-9.0/include/cudnn.h /usr/local/cuda-9.0/lib64/libcudnn*

 

 

 

おすすめ

転載: blog.csdn.net/u011939633/article/details/90758083