@@ -2,23 +2,28 @@ sudo apt update
2
2
sudo apt autoremove -y
3
3
sudo apt install software-properties-common
4
4
sudo add-apt-repository ppa:deadsnakes/ppa
5
- sudo apt install python3.8 -y
6
- # install cuda version 11.7
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
- wget https://developer.download.nvidia.com/compute/cuda/11.7.0/local_installers/cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
10
- sudo dpkg -i cuda-repo-ubuntu2004-11-7-local_11.7.0-515.43.04-1_amd64.deb
11
- sudo cp /var/cuda-repo-ubuntu2004-11-7-local/cuda-* -keyring.gpg /usr/share/keyrings/
5
+ sudo apt install python3.10 -y
6
+
7
+ # install cuda 12.3
8
+ # referece: https://developer.nvidia.com/cuda-downloads?target_os=Linux
9
+ wget https://developer.download.nvidia.com/compute/cuda/repos/ubuntu2204/x86_64/cuda-keyring_1.1-1_all.deb
10
+ sudo dpkg -i cuda-keyring_1.1-1_all.deb
11
+ sudo apt-get update
12
+ sudo apt-get -y install cuda-toolkit-12-3
13
+
12
14
sudo apt-get update
13
- # sudo apt-get -y install cuda
14
- # sudo apt -y install nvidia-cuda-toolkit
15
+ sudo apt-get -y install cuda
15
16
sudo add-apt-repository -y ppa:graphics-drivers/ppa && sudo apt update
17
+ sudo apt -y install nvidia-cuda-toolkit
18
+ sudo apt install ubuntu-drivers-common
16
19
sudo ubuntu-drivers autoinstall
17
20
sudo reboot
18
- # check nvidia-smi
21
+ # check nvidia-smi and nvcc
22
+ nvidia-smi
23
+ nvcc -V
24
+
19
25
# prepare torch environment
20
26
sudo apt install -y python3-pip
21
27
sudo pip3 install torch torchvision torchaudio
22
28
# install venv
23
- sudo apt install python3.8-venv
24
-
29
+ sudo apt install python3.10-venv
0 commit comments