ubuntu快速安装tensorflow\pytorch

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