原本的环境:windows10 + cuda7.5 + python3 + tensorflow1.1
1、卸载cuda7.5,windows下直接在控制面板卸载
ubuntu下,可以参考:http://blog.csdn.net/u012436149/article/details/53163346
2、卸载tensorflow: pip3 uninstall tensorflow
CUDA Installation Guide for Microsoft Windows
参考官方文档:http://docs.nvidia.com/cuda/cuda-installation-guide-microsoft-windows/index.html#axzz4TpI4c8vf%20
按照CUDA官网文档进行安装:
在Windows 7上安装CUDA8.0/CUDA7.5的系统需求:(1)、ACUDA-capable GPU(本机显卡为GeForce GT 640M);(2)、A supported version of Microsoft Windows(Windows7 64位);(3)、A supportedversion of Microsoft Visual Studio(已安装VS2013);(4)、the NVIDIACUDA Toolkit。
3、下载cuda8.0+cudnn8.0
https://developer.nvidia.com/cuda-downloads?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exelocal
4、安装显卡驱动,根据自己的显卡下载合适的
http://www.geforce.cn/drivers
5、安装cuda8.0
查看安装好的cuda,命令行:nvcc -V
6、安装cudnn8.0
(1)解压cudnn,会生成cuda/include、cuda/lib、cuda/bin三个目录;
(2)分别将cuda/include、cuda/lib、cuda/bin三个目录中的内容拷贝到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v8.0对应的include、lib、bin目录下即可。
7、安装tensorflow
pip3 install tensorflow-gpu