Skip to content

Instantly share code, notes, and snippets.

@bound2
Last active September 21, 2018 05:33
Show Gist options
  • Star 1 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save bound2/5f3912eb4f0abf8125459edd7cc87870 to your computer and use it in GitHub Desktop.
Save bound2/5f3912eb4f0abf8125459edd7cc87870 to your computer and use it in GitHub Desktop.
0-100 deepfake ubuntu linux setup
# 0-100 deepfake ubuntu setup 17.10 version:
# general stuff
sudo apt-get install git
sudo apt-get install cmake
# install nvidia drivers
sudo add-apt-repository ppa:graphics-drivers
sudo apt-get update
sudo apt-get purge nvidia-*
sudo apt-get install nvidia-390
sudo apt-mark hold nvidia-390
sudo sed -i 's/quiet splash/nomodeset/g' /etc/default/grub
sudo update-grub
reboot
# install cuda 9.0
wget https://developer.nvidia.com/compute/cuda/9.0/Prod/local_installers/cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64-deb
mv cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64-deb cuda-repo-ubuntu1704-9-0-local_9.0.176-1_amd64.deb
sudo apt-key add /var/cuda-repo-9-0-local/7fa2af80.pub
sudo apt-get update
sudo dpkg --configure -a
sudo apt-get install -f
# install tensorflow
sudo apt-get install python3-pip python3-dev python-virtualenv
mkdir ~/tensorflow
virtualenv --system-site-packages -p python3 tensorflow/
source ~/tensorflow/bin/activate
easy_install -U pip
pip3 install --upgrade tensorflow-gpu
# cudnn dependency - keep for backup https://developer.nvidia.com/compute/machine-learning/cudnn/secure/v7.0.4/prod/9.0_20171031/cudnn-9.0-linux-x64-v7
wget http://developer.download.nvidia.com/compute/redist/cudnn/v7.0.4/cudnn-9.0-linux-x64-v7.tgz
tar -xvzf cudnn-9.0-linux-x64-v7.tgz
sudo cp -P cuda/include/cudnn.h /usr/local/cuda/include
sudo cp -P cuda/lib64/libcudnn* /usr/local/cuda/lib64
sudo chmod a+r /usr/local/cuda/include/cudnn.h /usr/local/cuda/lib64/libcudnn*
# setup environment variables, with reboot
crontab -l | { cat; echo "@reboot export LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/cuda/extras/CUPTI/lib64"; } | crontab -
crontab -l | { cat; echo "@reboot export CUDA_HOME=/usr/local/cuda"; } | crontab -
crontab -l | { cat; echo "@reboot source ~/.bashrc"; } | crontab -
reboot
# install deepfake
source ~/tensorflow/bin/activate
mkdir ~/deepfakes
cd ~/deepfakes
git clone https://github.com/davisking/dlib
cd dlib
sudo python3 setup.py install --yes USE_AVX_INSTRUCTIONS --yes DLIB_USE_CUDA
pip3 install pathlib==1.0.1
pip3 install scandir==1.6
pip3 install h5py==2.7.1
pip3 install Keras==2.1.2
pip3 install opencv-python==3.3.0.10
pip3 install tensorflow-gpu==1.5.0
pip3 install scikit-image
pip3 install tqdm
pip3 install face_recognition
cd ..
git clone https://github.com/deepfakes/faceswap
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment