Keras在Linux平台的安装

前提是显卡驱动以及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  ## 注意查看运行信息

猜你喜欢

转载自blog.csdn.net/a18838956649/article/details/104832945