注意:tensorflow版本需要和机器上cuda和cudnn版本对应:
查看cuda版本:
cat /usr/local/cuda/version.txt
查看cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果需要安装对应版本:
conda install tensorflow=1.6.0
注意:tensorflow版本需要和机器上cuda和cudnn版本对应:
查看cuda版本:
cat /usr/local/cuda/version.txt
查看cudnn版本:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2
如果需要安装对应版本:
conda install tensorflow=1.6.0