azaion/ai-training
1
0
Fork 0
mirror of https://github.com/azaion/ai-training.git synced 2026-04-22 11:06:35 +00:00
Code Issues Packages Projects Releases Wiki Activity

small refactoring

Browse Source 
simplify learning, improve readme
This commit is contained in:
Alex Bezdieniezhnykh
2024-06-09 18:09:26 +03:00
parent a9b777acc4
commit 70abaded05
2 changed files with 56 additions and 50 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+14 -2
View File
@@ -1,8 +1,20 @@
1. Install dependencies first
1. Install necessary libs and drivers
```
sudo apt install nvidia-driver-535
wget https://developer.download.nvidia.com/compute/cudnn/9.2.0/local_installers/cudnn-local-repo-ubuntu2204-9.2.0_1.0-1_amd64.deb
sudo dpkg -i cudnn-local-repo-ubuntu2204-9.2.0_1.0-1_amd64.deb
sudo cp /var/cudnn-local-repo-ubuntu2204-9.2.0/cudnn-*-keyring.gpg /usr/share/keyrings/
sudo apt-get update
sudo apt-get -y install cudnn nvidia-cuda-toolkit -y
nvcc --version
```
2. Install dependencies
```
python -m pip install --upgrade pip
pip install --upgrade huggingface_hub
pip install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip install -q git+https://github.com/THU-MIG/yolov10.git
pip install albumentations
```
```