nvidia

elwood-software · Dec 11, 2024 · f84af30 · f84af30
1 parent b681849
commit f84af30
Show file tree

Hide file tree

Showing 3 changed files with 11 additions and 20 deletions.
diff --git a/build/images.ts b/build/images.ts
@@ -22,7 +22,7 @@ const images: Image[] = [
   {
     source_ami: "ami-0453ec754f44f9a4a",
     arch: "nvidia",
-    instance_type: "p2.xlarge",
+    instance_type: "p3.2xlarge",
     region: "us-east-1",
   },
 ];

diff --git a/build/install-ffmpeg-nvidia.sh b/build/install-ffmpeg-nvidia.sh
@@ -2,6 +2,9 @@
 #
 # adapted from https://gist.github.com/bilalmughal/cb89936bc947fa727a8ec66e3ddf768a
 #
+# https://www.nvidia.com/en-us/drivers/
+# https://developer.nvidia.com/cuda-downloads
+# https://developer.nvidia.com/cudnn-downloads?target_os=Linux
 
 set -e # Exit on any error
 
@@ -82,7 +85,7 @@ install_utils() {
     echo "Installing packages..."
     if [ "$package_manager" = "dnf" ]; then
         $package_manager -y groupinstall "Development Tools"
-        $package_manager install -y git autoconf openssl-devel cmake3 htop iotop yasm nasm jq freetype-devel fribidi-devel harfbuzz-devel fontconfig-devel bzip2-devel kernel-modules-extra
+        $package_manager install -y git autoconf kernel-modules-extra openssl-devel cmake3 htop iotop yasm nasm jq freetype-devel fribidi-devel harfbuzz-devel fontconfig-devel bzip2-devel
     elif [ "$package_manager" = "apt" ]; then
         export DEBIAN_FRONTEND=noninteractive
         export NEEDRESTART_MODE=a
@@ -104,13 +107,13 @@ setup_gpu() {
     fi
     if [ "$(uname -m)" = "aarch64" ]; then
         echo "System is running on ARM / AArch64"
-        DRIVE_URL="https://us.download.nvidia.com/tesla/535.104.05/NVIDIA-Linux-aarch64-535.104.05.run"
-        CUDA_SDK_URL="https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux_sbsa.run"
-        CUDNN_ARCHIVE_URL="https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-sbsa/cudnn-linux-sbsa-8.9.5.29_cuda12-archive.tar.xz"
+        DRIVE_URL="https://us.download.nvidia.com/tesla/550.127.08/NVIDIA-Linux-aarch64-550.127.08.run"
+        CUDA_SDK_URL="https://developer.download.nvidia.com/compute/cuda/12.6.3/local_installers/cuda_12.6.3_560.35.05_linux_sbsa.run"
+        CUDNN_ARCHIVE_URL="https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-sbsa/cudnn-linux-sbsa-9.6.0.74_cuda12-archive.tar.xz"
     else
-        DRIVE_URL="https://us.download.nvidia.com/tesla/535.104.05/NVIDIA-Linux-x86_64-535.104.05.run"
-        CUDA_SDK_URL="https://developer.download.nvidia.com/compute/cuda/12.2.2/local_installers/cuda_12.2.2_535.104.05_linux.run"
-        CUDNN_ARCHIVE_URL="https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-8.9.5.29_cuda12-archive.tar.xz"
+        DRIVE_URL="https://us.download.nvidia.com/tesla/550.127.08/NVIDIA-Linux-x86_64-550.127.08.run"
+        CUDA_SDK_URL="https://developer.download.nvidia.com/compute/cuda/12.6.3/local_installers/cuda_12.6.3_560.35.05_linux.run"
+        CUDNN_ARCHIVE_URL="https://developer.download.nvidia.com/compute/cudnn/redist/cudnn/linux-x86_64/cudnn-linux-x86_64-9.6.0.74_cuda12-archive.tar.xz"
     fi
 
     echo "Setting up GPU..."
@@ -311,4 +314,3 @@ install_ffmpeg_prereqs
 install_ffmpeg
 popd
 rm -fr $SRC_DIR
-/usr/local/bin/ffmpeg -version >/tmp/version.txt
diff --git a/docs/build-data/version-nvidia.txt b/docs/build-data/version-nvidia.txt
@@ -1,11 +0,0 @@
-ffmpeg version N-118030-g550c662a1f Copyright (c) 2000-2024 the FFmpeg developers
-built with gcc 11 (GCC)
-configuration: --prefix=/usr/local --disable-static --enable-shared --extra-cflags='-I/usr/local/include -I/usr/local/cuda/include' --extra-ldflags='-L/usr/local/lib -L/usr/local/cuda/lib64' --extra-libs='-lpthread -lm' --bindir=/usr/local/bin --enable-gpl --enable-libaom --enable-libass --enable-libfdk-aac --enable-libfreetype --enable-libmp3lame --enable-libopus --enable-libvorbis --enable-libvpx --enable-libx264 --enable-libx265 --enable-nonfree --enable-openssl --enable-cuda-nvcc --nvcc=/usr/local/cuda/bin/nvcc --enable-cuda --enable-cuvid --enable-nvenc
-libavutil      59. 49.100 / 59. 49.100
-libavcodec     61. 26.100 / 61. 26.100
-libavformat    61.  9.100 / 61.  9.100
-libavdevice    61.  4.100 / 61.  4.100
-libavfilter    10.  6.101 / 10.  6.101
-libswscale      8. 12.100 /  8. 12.100
-libswresample   5.  4.100 /  5.  4.100
-libpostproc    58.  4.100 / 58.  4.100