Prerequisite: conda and/or miniconda are already installed
- Create a conda environment.
$ conda create -n dlib python=3.8 cmake ipython- Activate the environment.
$ conda activate dlib- Install CUDA and cuDNN with
condausing nvidia channel
$ conda install cuda cudnn -c nvidiaThen find the path to the nvcc of this environment. We will use this path for the build step below
$which nvcc
/path/to/your/miniconda3/envs/dlib/bin/- Install dlib. Clone and build dlib from source
$ git clone https://github.com/davisking/dlib.git
$ cd dlib
$ mkdir build
$ cd build
$ cmake .. -DDLIB_USE_CUDA=1 -DUSE_AVX_INSTRUCTIONS=1 -DCUDAToolkit_ROOT=/path/to/your/miniconda3/envs/dlib/bin/
$ cmake --build .
$ cd ..
$ python setup.py install --set DLIB_USE_CUDA=1- Test dlib
(dlib) $ ipython
Python 3.8.12 (default, Oct 12 2021, 13:49:34)
Type 'copyright', 'credits' or 'license' for more information
IPython 7.27.0 -- An enhanced Interactive Python. Type '?' for help.
In [1]: import dlib
In [2]: dlib.DLIB_USE_CUDA
Out[2]: True
In [3]: print(dlib.cuda.get_num_devices())
1
hi, thank you for the fast response, i will surely try, but i've tried also to install pytorch from conda and encountered the same problem:
$ conda install pytorch==1.13.1 torchvision==0.14.1 torchaudio==0.13.1 -c pytorch
Channels:
Platform: linux-64
Collecting package metadata (repodata.json): done
Solving environment: failed
InvalidSpec: The package "nvidia/linux-64::cuda-compiler==12.6.0=0" is not available for the specified platform
do you think that i can install all from pip? i'm worried of installing from pip because i've a lot of dependencies.
Thank you again