sudo apt update
sudo apt install python3-dev python3-pip
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ tensorflow-gpu
pip3 install -i https://pypi.tuna.tsinghua.edu.cn/simple/ torch torchvision
若tensorflow出现报错：ImportError: libcublas.so.9.0: cannot open shared object file: No such file or directory
在正确安装cuda和cudnn的情况下，执行一下命令即可
sudo ldconfig /usr/local/cuda-9.0/lib64
