gpu版本tensorflow及pytorch安装相关

1、tensorflow gpu版本好用的安装命令

pip方式安装命令

pip install tensorflow-gpu==1.5.0；

2、facebook detectron 安装过程记录

github 地址：https://github.com/facebookresearch/Detectron

电脑环境：Ubuntu14.04 ，gpu GTX970，cuda8.0, cudnn6.0.22

anaconda2虚拟环境下安装：

创建虚拟环境：anaconda create -n caf2 python==2.7.14;

下载 pytorch安装包；

安装过程参考链接：https://caffe2.ai/docs/getting-started.html?platform=mac&configuration=compile

依赖项安装：conda install -y     future     gflags     glog     leveldb     mkl     mkl-include     numpy     opencv     protobuf     six

cd ~path/pytorch/build

cmake -DCMAKE_PREFIX_PATH=~/anaconda2/envs/caf2 -DCMAKE_INSTALL_PREFIX=~/anaconda2/envs/caf2 ..

make install -j8

安装后测试：

cd ~ && python -c 'from caffe2.python import core' 2>/dev/null && echo "Success" || echo "Failure"

3、cuda及cudnn版本查询

https://medium.com/@changrongko/nv-how-to-check-cuda-and-cudnn-version-e05aa21daf6c

cuda 版本
cat /usr/local/cuda/version.txt

cudnn 版本
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2