Last active
February 18, 2020 17:40
-
-
Save tldrafael/8e31be5ce0f55f8412e63a3270f9ebaa to your computer and use it in GitHub Desktop.
Configuring EC2 AWS g4 GPUs machine
This file contains hidden or bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| # 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