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

fix readme

Browse Source 
add export py
add dll
This commit is contained in:
Alex Bezdieniezhnykh
2024-08-23 13:40:08 +03:00
parent d2dae732b9
commit 330d4b914a
7 changed files with 57 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
README.md
+41 -12
View File
@@ -1,19 +1,48 @@
1. Install necessary libs and drivers
```
sudo apt install nvidia-driver-535
**1. Install necessary libs and drivers**
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
https://www.python.org/downloads/
Windows
* [Install CUDA](https://developer.nvidia.com/cuda-12-1-0-download-archive)
Linux
* ```
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**
* (Optional) Install virtual env:
```
pip install virtualenv
python -m venv env
env\Scripts\activate
```
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 ultralytics albumentations
pip install git+https://github.com/airockchip/ultralytics_yolov8.git
pip install albumentations onnx
```
**3. Fix possible problems**
* Windows:
* cv2.error: OpenCV(4.10.0) ...\window.cpp:1301: error: (-2:Unspecified error)
```
pip uninstall opencv-python
pip install opencv-python
```
* fbgemm.dll error
```
copypaste libomp140.x86_64.dll to C:\Windows\System32
```