A-A+

最新tensorflow2 安装与环境配置

2020年05月23日 23:04 学习笔记 暂无评论 共1417字 (阅读1,656 views次)

【注意:此文章为博主原创文章!转载需注意,请带原文链接,至少也要是txt格式!】

安装cuda

第一步要安装cuda 下载地址:https://developer.nvidia.com/cuda-downloads

这里要注意你N卡的版本。例如我的版本是:

Nvidia 驱动版本

我的是441.22版本。这里要注意一下,电脑是否安装微软的vscode,如果安装了就需要特殊的设置,否则不需要。安装cuda 时请选择自选安装。

去掉多余cuda选项

去掉cuda VScode选项

注意cuda版本与驱动版本对应信息(注意,如果相等,可以去掉此选项)

多个cuda版本是可以共存的,所以如果你安装了很多版本是没关系的,具体使用哪个版本取决你系统环境变量的设置。

cuda windows系统环境变量配置

cuda windows系统环境变量配置

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 可以直接点击连接查看解决方案。

布施恩德可便相知重

微信扫一扫打赏

支付宝扫一扫打赏

×
标签:

给我留言