Windows下安装TensorFlow-gpu

来源：互联网发布：刮骨疗伤是真的吗知乎编辑：程序博客网时间：2024/05/22 07:09

在之前需要安装CUDA Toolkit 8.0.
https://pan.baidu.com/s/1c2y8xIK和cuDNN v6 or v6.1。
一定要版本正确，否则会出现问题，在官网上有说明。

直接一直点击确认即可，安装后可以看到系统中多了CUDA_PATH和CUDA_PATH_V8_0两个环境变量，接下来要在系统中添加以下几个环境变量： 　　CUDA_SDK_PATH=C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0　　 CUDA_LIB_PATH=%CUDA_PATH%\lib\x64 　　CUDA_BIN_PATH=%CUDA_PATH%\bin 　　CUDA_SDK_BIN_PATH=%CUDA_SDK_PATH%\bin\win64 　　CUDA_SDK_LIB_PATH=%CUDA_SDK_PATH%\common\lib\x64 　　然后，在系统变量PATH的末尾依次添加： 　　%CUDA_LIB_PATH%　　%CUDA_BIN_PATH%　　%CUDA_SDK_LIB_PATH%　　%CUDA_SDK_BIN_PATH%　　说明：这里注意的是CUDA_SDK_PATH路径的位置，很多人去这个错误路径“C:\Program Files\NVIDIA Corporation”下，发现找不到CUDA Samples文件夹。实际上是因为ProgramData是一个隐藏文件夹，需要设置【我的电脑】-【查看】-【显示隐藏文件】才能看到，因此这个才是正确路径“C:\ProgramData\NVIDIA Corporation\CUDA Samples\v8.0”。　　配置成功后，重启计算机使环境变量生效。

conda create -n tensorflow python=3.5
activate tensorflow
pip install –ignore-installed –upgrade tensorflow-gpu

阅读全文

0 0