前提是显卡驱动以及CUDA cuDNN提前装好
可以用以下代码进行测试
nvidia-smi # 查看显卡驱动是否安装好
nvcc -V # 查看cuDNN是否装好
接下来正式安装Keras\TensorFlow
conda crtate -n test python=3.6 # 注意是3.6版本,创建环境名为test的环境
conda activate test # 激活test环境
conda install -c anaconda tensorflow-gpu
conda install -c anaconda keras-gpu
conda install -c anaconda h5py graphviz pydot
配置Keras后端,由于默认后端是TensorFlow,因此我们不必配置
接下来测试是否安装成功
git clone https://github.com/keras-team/keras.git
cd keras/examples
python mnist_cnn.py ## 注意查看运行信息