NVideo

<!-- install -->
sudo apt update
sudo apt install nvidia-driver-550

<!-- check -->
sudo find /usr -name "libcuda.so*" 2>/dev/null

<!-- symlink -->
sudo ln -s /usr/lib/x86_64-linux-gnu/libcuda.so.1 /usr/lib/x86_64-linux-gnu/libcuda.so

<!-- Toolkit -->
export LIBRARY_PATH=/usr/lib/x86_64-linux-gnu:$LIBRARY_PATH

<!-- nvidea -->
sudo apt install nvidia-cuda-toolkit


pip install torch torchvision --index-url https://download.pytorch.org/whl/cu126

python -c "import torch; print(torch.version.cuda); print(torch.cuda.is_available())"

Environment

sudo add-apt-repository ppa:deadsnakes/ppa
sudo apt update
sudo apt install python3.11 python3.11-venv

pip install uv
python3.11 -m venv .venv
. .venv/bin/activate

pip install -r requirements.txt

Make zip file

zip -r fine_tuning.zip .  -x "__pycache__/*" ".hypothesis/*" ".venv/*" "dataset/*" "utils/__pycache__/*" 

GPU Check

nvidia-smi