碎碎念
前段时间电脑出了点问题,换了个屏幕而且重装了。这个暑期的项目不多,但是想跟着github上的大佬学习一波。以前的代码都随便乱扔的,相当于没有任何印象,因此希望通过培养发博客这个习惯给自己整理一下。
当前环境
- 系统 :WIN10
- python版本:3.7.3 用Anaconda直接装
- Visual Studio:2017(目前只有2017有社区版并且支持cuda)
安装
安装tensorflow,在cmd直接输入pip install --upgrade tensorflow,等着即可。
接着下载 pip install --upgrade tensorflow-gpu 安装gpu版本的tensorflow。
从cmd进入python以后,输入
import tensorflow
# 提示
ImportError: Could not find 'cudart64_100.dll'. TensorFlow requires that this DLL be installed in a directory
that is named in your %PATH% environment variable. Download and install CUDA 10.0 from this URL: https://devel
oper.nvidia.com/cuda-90-download-archive
差个10.0的cuda和cudnn(这个需要注册英伟达账号)
点击进入选择相应的版本下载,对应上面报错的信息,我选择的是10.0的cuda, cudnn要记得勾选以后才可显示下载列表噢!
安装cuda过程中,要保证有vs2017。
安装选项有精简和自定义~~(当然是自定义比精简更精简)~~ ,由于重装过,我的C盘还是有点空间的,所以就选了精简,把全家桶给弄下来了。
安装位置最好选到C盘吧。
安装的时候一般都会帮你自动把路径加入到环境变量中。
找了一下网上的办法,在cmd输入
nvcc -V
说明CUDA安装成功了!
接下来继续把cuDNN给整好,解压下载下来的文件后,有这几个目录
- bin
- include
- lib
- NVIDIA_SLA_cuDNN_Support.txt
按照网上的教程,直接把这几个目录复制到CUDA目录下即可。(如果觉得不够稳,可以把cuDNN中的文件一个个地复制到CUDA相应的文件夹中)
然后!!!
我又重新在python中输入了import tensorflow
结果还是一个样???
认真看了看这个错误,说的是在你的环境变量的目录下找不到cudart64_100.dll
这个文件。于是,到CUDA的bin目录下一找
这不是好好的吗?
再看了一下环境变量,果然出了点小问题
CUDA所在路径\CUDA\v10.0\bin
CUDA所在路径\CUDA\v10.0\lib\x64\
改成以上两个即可
再来import tensorflow