본문 바로가기
IT 와 Social 이야기/Python

[Tech With Tim] How to Install TensorFlow GPU on Linux

by manga0713 2020. 6. 17.






*** Python Neural Networks with TensorFlow 9

: How to Install TensorFlow GPU on Linux


# Add NVIDIA package repositories wget https://developer.download.nvidia.com... sudo dpkg -i cuda-repo-ubuntu1804_10.0.130-1_amd64.deb sudo apt-key adv --fetch-keys https://developer.download.nvidia.com... sudo apt-get update wget http://developer.download.nvidia.com/... sudo apt install ./nvidia-machine-learning-repo-ubuntu1804_1.0.0-1_amd64.deb sudo apt-get update # Install NVIDIA driver sudo apt-get install --no-install-recommends nvidia-driver-410 # Reboot. Check that GPUs are visible using the command: nvidia-smi # Install development and runtime libraries (~4GB) sudo apt-get install --no-install-recommends \ cuda-10-0 \ libcudnn7=7.4.1.5-1+cuda10.0 \ libcudnn7-dev=7.4.1.5-1+cuda10.0 # Install TensorRT. Requires that libcudnn7 is installed above. sudo apt-get update && \ sudo apt-get install nvinfer-runtime-trt-repo-ubuntu1804-5.0.2-ga-cuda10.0 \ && sudo apt-get update \ 

&& sudo apt-get install -y --no-install-recommends libnvinfer-dev=5.0.2-1+cuda10.0