機器學習相關

  • 環境初始化

    cuda: http://developer.download.nvidia.com/compute/cuda/repos/rhel7/x86_64/
    cuDNN:http://developer.download.nvidia.com/compute/redist/cudnn/v5.1/cudnn-8.0-linux-x64-v5.1.tgz
    http://developer.download.nvidia.com/compute/redist/cudnn/v6.0/cudnn-8.0-linux-x64-v6.0.tgz
    http://developer.download.nvidia.com/compute/redist/cudnn/v7.0.5/cudnn-8.0-linux-x64-v7.tgz
    http://developer.download.nvidia.com/compute/redist/cudnn/v7.0.5/cudnn-9.0-linux-x64-v7.tgz

    • 查看 cuda 版本:cat /usr/local/cuda/version.txt
    • 查看 cudnn 版本:cat /usr/local/cuda/include/cudnn.h | grep 'define CUDNN_MAJOR' -A 2

  • tensorflow

    bazel: https://copr.fedorainfracloud.org/coprs/vbatts/bazel/repo/epel-7/vbatts-bazel-epel-7.repo
    https://www.variantweb.net/pub/bazel-0.5.2-1.fc26.src.rpm
    tensoflow: git clone --recursive https://github.com/tensorflow/tensorflow && cd tensoflow && git checkout v1.4.1

    • 版本依賴
    tensoflow 1.4 with cuda 8.0 and cuDNN 6.0
    tensoflow 1.5 with cuda 9.0 and cuDNN 7.0

    發表評論
    所有評論
    還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
    相關文章