Skip to content

Instantly share code, notes, and snippets.

@bunsyy
Last active February 18, 2025 03:28
Show Gist options
  • Save bunsyy/a471dc73c10f5858789cec002e1619ea to your computer and use it in GitHub Desktop.
Save bunsyy/a471dc73c10f5858789cec002e1619ea to your computer and use it in GitHub Desktop.
sudo apt update && sudo apt upgrade
sudo apt autoremove nvidia* --purge
ubuntu-drivers devices
sudo ubuntu-drivers autoinstall
sudo apt install nvidia-driver-570
sudo apt update && sudo apt upgrade
sudo apt install nvidia-cuda-toolkit
nvcc --version
sudo apt-get install python3-pip
sudo pip3 install virtualenv
virtualenv -p python3.10 venv
source venv/bin/activate
pip3 install torch torchvision torchaudio
import torch
print(torch.cuda.is_available()) # Should return True
t = torch.rand(10, 10).cuda()
print(t.device) # Should return 'cuda'
sudo apt install ./<filename.deb>
sudo cp /var/cudnn-<something>.gpg /usr/share/keyrings/
sudo apt update
sudo apt install libcudnn8
sudo apt install libcudnn8-dev
sudo apt install libcudnn8-samples
sudo apt install ./cudnn-local-repo-cross-sbsa-ubuntu2204-9.7.1_1.0-1_all.deb
sudo apt install ./<filename.deb>
sudo apt update
sudo apt install tensorrt
trtexec: /usr/src/tensorrt/bin/trtexec
whereis trtexec
/usr/src/tensorrt/bin/trtexec
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment