1, cuda10.1
tf2.0 不支持cuda10.1,用conda重新安装cuda和cudnn:
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/cudnn-7.6.4-cuda10.0_0.tar.bz2
wget https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/linux-64/cudatoolkit-10.0.130-0.tar.bz2
conda install cudatoolkit-10.0.130-0.tar.bz2
conda install cudnn-7.6.4-cuda10.0_0.tar.bz2
cudnn >= 7.6,否则会报错:
Failed to get convolution algorithm. This is probably because cuDNN failed to initialize, so try looking to see if a warning log message was printed above. [Op:Conv2D]
验证:
import tensorflow as tf
len(tf.config.experimental.list_physical_devices('GPU')),len(tf.config.experimental.list_logical_devices('GPU'))
2,juypter环境
source activate [conda-env]
conda install ipykernel
python -m ipykernel install --user --name [conda-env]