File tree Expand file tree Collapse file tree 1 file changed +1
-11
lines changed Expand file tree Collapse file tree 1 file changed +1
-11
lines changed Original file line number Diff line number Diff line change 11#! /bin/bash
22
33echo " Running dependency installation for Ubuntu."
4-
5- mkdir ubuntu_temp
6- pushd ubuntu_temp || exit
7- wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/cuda-ubuntu2004.pin
8- sudo mv cuda-ubuntu2004.pin /etc/apt/preferences.d/cuda-repository-pin-600
9- sudo apt-key adv --fetch-keys https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/3bf863cc.pub
10- sudo add-apt-repository " deb https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2004/x86_64/ /"
11- sudo apt-get update
12- popd || exit
13-
14- apt -y install nvidia-cuda-dev nvidia-cuda-toolkit cuda-cupti-11-7 libnccl-dev cuda-11-7
4+ apt -y install nvidia-cuda-dev nvidia-cuda-toolkit cuda-cupti-11-7 libnccl-dev
155echo " Creating python venv"
166python -m venv .venv
177echo " Activating venv"
You can’t perform that action at this time.
0 commit comments