Skip to content

Instantly share code, notes, and snippets.

@tldrafael

tldrafael/CONFIGURE

Last active February 18, 2020 17:40
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save tldrafael/8e31be5ce0f55f8412e63a3270f9ebaa to your computer and use it in GitHub Desktop.
Save tldrafael/8e31be5ce0f55f8412e63a3270f9ebaa to your computer and use it in GitHub Desktop.
Download ZIP
Configuring EC2 AWS g4 GPUs machine
Raw
CONFIGURE
# Install cuda-10.0 because the newest one (for now the version 10.2) mismatchs the tensorflow 1.14.0 requirements
wget https://developer.nvidia.com/compute/cuda/10.0/Prod/local_installers/cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd6
dpkg -i cuda-repo-ubuntu1804-10-0-local-10.0.130-410.48_1.0-1_amd6
apt-key add /var/cuda-repo-10-0-local-10.0.130-410.48/7fa2af80.pub
apt-get install cuda
# This cuda version install nvidia v410 drivers. Remove it.
apt-get remove --purge nvidia-*
# The G4 instances need at least driver version 426 as said https://docs.aws.amazon.com/AWSEC2/latest/UserGuide/install-nvidia-driver.html
apt-get install nvidia-driver-435
# Reboot machine
# Manually access the page https://developer.nvidia.com/rdp/cudnn-download, login required
# Download cuDNN v7.6.5 (November 5th, 2019), for CUDA 10.0
dpkg -i libcudnn7_7.6.5.32-1+cuda10.0_amd64.deb
apt-get install libcudnn7
# Now it is ready to go!
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment