Skip to content

Commit

Permalink
chore: Updated an example of installing cuda12 and cudnn9 on ubuntu 2…
Browse files Browse the repository at this point in the history
…2.04
  • Loading branch information
hh-space-invader committed Sep 19, 2024
1 parent 3a5133a commit 466badc
Showing 1 changed file with 22 additions and 3 deletions.
25 changes: 22 additions & 3 deletions docs/examples/FastEmbed_GPU.ipynb
Original file line number Diff line number Diff line change
Expand Up @@ -93,14 +93,33 @@
"\n",
"#### CUDA drivers\n",
"You can download the CUDA 11.8 toolkit from the NVIDIA website found [here](https://developer.nvidia.com/cuda-11-8-0-download-archive) and CUDA 12.x [here](https://developer.nvidia.com/cuda-downloads) if not already preinstalled.\n",
"#### CuDNN 8.x\n",
"You can download the CuDNN 8.x library from the NVIDIA website found [here](https://developer.nvidia.com/rdp/cudnn-archive)\n",
"#### Example of setting up CUDA 12.x on Ubuntu 22.04\n",
"[Here](https://developer.nvidia.com/cuda-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_network) is a selected choice for platform (Linux), then CPU archetecture (x86_64), then OS distribution (Ubuntu), then version (22.04), and then installer type (deb_network):\n",
"```bash\n",
"wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb\n",
"sudo dpkg -i cuda-keyring_1.1-1_all.deb\n",
"sudo apt-get update\n",
"sudo apt-get -y install cuda\n",
"```\n",
"**NOTE**: You can install specific libraries of CUDA by checking [here](https://docs.nvidia.com/cuda/cuda-installation-guide-linux/#meta-packages).\n",
"\n",
"**NOTE**: When installing CUDA, the environment variable might not be set by default. Make sure to add the following line to your environment variables:\n",
"```bash\n",
"LD_LIBRARY_PATH=/usr/local/cuda/lib64:$LD_LIBRARY_PATH\n",
"```\n",
"This will ensure that the CUDA libraries are properly linked."
"This will ensure that the CUDA libraries are properly linked.\n",
"\n",
"#### CuDNN 9.x\n",
"You can download the CuDNN 9.x library from the NVIDIA website found [here](https://developer.nvidia.com/rdp/cudnn-archive)\n",
"#### Example of setting up CuDNN 9.x on Ubuntu 22.04\n",
"[Here](https://developer.nvidia.com/cudnn-downloads?target_os=Linux&target_arch=x86_64&Distribution=Ubuntu&target_version=22.04&target_type=deb_network) is the same concept as downloading CUDA.\n",
"```bash\n",
"wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb\n",
"sudo dpkg -i cuda-keyring_1.1-1_all.deb\n",
"sudo apt-get update\n",
"sudo apt-get -y install cudnn\n",
"```\n",
"**NOTE**: When installing CuDNN, you can choose specific version, cudnn-cuda-11 or cudnn-cuda-12"
]
},
{
Expand Down

0 comments on commit 466badc

Please sign in to comment.