以TensorFlow1.14-gpu為例,服務器先安裝好docker服務,以及顯卡驅(qū)動,這個的好處是只需要自己安裝驅(qū)動即可實用,不需要安裝cuda和cudnn。
1、下載docker image
docker pull tensorflow/tensorflow:1.14.0-gpu-py3
2、帶gpu運行docker,掛載本地目錄,假設代碼在本地目錄/home/song/develop_project/my_project_1中,掛載到docker的/home/my_project_1中:
run -itd --name tensorflow_14 --gpus all -v /home/song/develop_project/my_project_1://home/my_project_1 tensorflow/tensorflow:1.14.0-gpu-py3 /bin/bash
3、進入docker
docker exec -it tensorflow_14 /bin/bash
4、執(zhí)行更新
apt-get update
5、安裝依賴包
apt-get install libpq-dev
6、更新pip,通過豆瓣更新。
pip install --upgrade pip -i http://pypi.douban.com/simple --trusted-host pypi.douban.com
8、安裝項目需要的包,通過豆瓣更新。
pip install -r requirements.txt -i http://pypi.douban.com/simple --trusted-host pypi.douban.com