SAEHD refuses to use my GPU

Open MarkusMaribu opened this issue 4 years ago • 3 comments

I have a GTX 1070. It recognizes it in windows, but on the linux build, it's as if DeepFaceLab only recognizes my CPU. Im kinda a linux noob, so be gentle.

Screenshot from 2021-11-22 21-13-21

Nov 22 '21 20:11 MarkusMaribu

#52 hope this helps

Nov 29 '21 05:11 th3j0ker

Downgrade Tensorflow

Mar 01 '22 01:03 fungtion

install env conda create -n deepfacelab -c main python=3.7 cudnn=7.6.5 cudatoolkit=10.1.243

my requirements file DeepFaceLab/requirements-cuda.txt

tqdm
numpy==1.16.1
numexpr
h5py==2.10.0
opencv-python==4.1.0.25
ffmpeg-python==0.1.17
scikit-image==0.14.2
scipy==1.4.1
colorama
tensorflow-gpu==2.3.1
pyqt5

Mar 11 '24 07:03 chwshuang