mirror of
https://github.com/azaion/ai-training.git
synced 2026-04-22 22:16:35 +00:00
Alex Bezdieniezhnykh
41 lines
1.1 KiB
Markdown
41 lines
1.1 KiB
Markdown
**1. Install necessary libs and drivers**
|
|
|
|
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
|
|
pip install -r requirements.txt
|
|
```
|
|
|
|
**3. Fix possible problems**
|
|
* cv2.error: OpenCV(4.10.0) ...\window.cpp:1301: error: (-2:Unspecified error)
|
|
```
|
|
pip uninstall opencv-python
|
|
pip install opencv-python
|
|
```
|
|
* fbgemm.dll error (Windows specific)
|
|
```
|
|
copypaste tests\libomp140.x86_64.dll to C:\Windows\System32
|
|
```
|