- numpy(我是1.18.1)
import numpy
numpy.__version__
- python版本(3.7.6)
- keras
pip install keras==2.2.4 -i https://pypi.doubanio.com/simple
- tensorflow(1.15.0)
import tensorflow as tf
tf.__version__
pip install tensorflow-gpu==1.15.0 -i https://pypi.doubanio.com/simple
查看cuda的版本
cmd 输入命令:
nvcc --version
nvcc: NVIDIA (R) Cuda compiler driver
Copyright (c) 2005-2018 NVIDIA Corporation
Built on Sat_Aug_25_21:08:04_Central_Daylight_Time_2018
Cuda compilation tools, release 10.0, V10.0.130
可以看到我的cuda版本是10.0
2 查看cudnn版本
进入目录查看
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v9.0\include\cudnn.h
文件内容如下:
可以看出版本是 7.6.5
3 安装numpy
pip install numpy==1.18.1 -i https://pypi.tuna.tsinghua.edu.cn/simple