tensorflow2.0-gpu install note

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]

猜你喜欢

转载自blog.csdn.net/dragonchow123/article/details/103793787
今日推荐