最新tensorflow2 安装与环境配置

【注意:此文章为博主原创文章!转载需注意,请带原文链接,至少也要是txt格式!】
安装cuda
第一步要安装cuda 下载地址:https://developer.nvidia.com/cuda-downloads
这里要注意你N卡的版本。例如我的版本是:
我的是441.22版本。这里要注意一下,电脑是否安装微软的vscode,如果安装了就需要特殊的设置,否则不需要。安装cuda 时请选择自选安装。
多个cuda版本是可以共存的,所以如果你安装了很多版本是没关系的,具体使用哪个版本取决你系统环境变量的设置。
cuda windows系统环境变量配置
为了方便直接给出路径:
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\bin
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\libnvvp
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\extras\CUPTI\lib64
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\cudnn\bin
具体是否重启看系统提示,这里要注意这个路径,这个是不存在的,需要我们自己创建并放进去cudnn文件。
这里解压cudnn-10.1-windows10-x64-v7.6.5.32.zip ,然后改名文件夹为cudnn,然后放入到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.2\目录下即可。
在cmd下执行命令nvcc -V可以查看当前cuda安装显示的版本和你安装的是否一致。
安装anaconda3
这个双击运行就好,安装后注意环境变量的配置,上一步图片中蓝色已经花圈了,添加上就OK了。
安装tensorflow2
首先升级pip
python -m pip install --upgrade
这里,如果要安装tensorflow需要翻墙,步然超级慢的哦。
pip install tensorflow
pip install tensorflow-gpu
如果使用过程中遇到了:Could not load dynamic library ‘cudart64_101.dll’; dlerror: cudart64_101.dll not found 可以直接点击连接查看解决方案。
布施恩德可便相知重
微信扫一扫打赏
支付宝扫一扫打赏