-
Notifications
You must be signed in to change notification settings - Fork 163
Install CUDA
Joongi Kim edited this page Jan 2, 2018
·
3 revisions
On the clouds, we highly recommend using vendor-provided GPU-optimized instance types (e.g., p2/p3 series on AWS) and GPU-optimized virtual machine images which include ready-to-use CUDA drivers and configurations.
Since Backend.AI's kernel container images ship all the necessary libraries and 3rd-party computation packages, you may choose the light-weight "base" image (e.g., Amazon Deep Learning Base AMI) instead of full-featured images (e.g., Amazon Deep Learning Conda AMI).
Please search for this topic on the Internet, as Linux distributions often provide their own driver packages and optimized method to install CUDA.
To download the driver and CUDA toolkit directly from NVIDIA, visit here.