Win10系统安装Tensorflow-GPU和VSCode构建Tensorflow开发环境

news/2024/11/30 10:47:47/

【前言】
1. 最近因为上课需要安装Anaconda和Tensorflow-GPU,Anaconda安装很容易,但Tensorflow-GPU版本的安装较为复杂,因为需要考虑版本匹配的一些问题,很容易出现问题。我花了两天多的时间终于安装成功,属实不易,下面分享一下我的安装过程
2. 安装环境
操作系统:64位Win10
GPU:MX130
IDE:VS Code
其他:CUDA10.1、cuDNN7.6.5


【安装过程】
1.安装Anaconda
版本:2020.07
百度网盘下载链接 提取码:k8p0
安装较为简单,这里只放几张较为重要的图片
在这里插入图片描述

在这里插入图片描述
在这里插入图片描述

2.安装CUDA10.1
CUDA是基于C语言的,可以让代码直接在GPU中运行的控制语言,所以第一步需要安装CUDA10.1,内含NVIDIA驱动程序(自定义安装时可以选择是否安装驱动)。
2.1 CUDA官方说明(感兴趣可以看一下)
下载地址:点这里
百度网盘下载链接 提取码:e06o
在这里插入图片描述

选择适合自己电脑的版本下载,exe是可执行文件。

2.2安装CUDA
几个文件下载好之后挨个点击进行安装即可,会先检查系统兼容性,然后选择安装方式。程序默认的精简模式应该可以理解为安装所有东西,其中包括了我暂时不用的VS编译器和显卡驱动,所以我选择的是自定义模式。

在这里插入图片描述
在这里插入图片描述
接下来是选择安装路径。在其他同学的安装过程中发现更改安装路径容易出现一些问题,所以我就直接按照默认路径进行安装的,这样也便于下面配置环境变量,一路Next 直到完成即可。
在这里插入图片描述

2.3配置环境变量
安装成功后在系统的环境变量中会自动添加两个变量,如图:
在这里插入图片描述

但是这还不够,我们还需要再添加几个系统变量(根据实际安装版本和路径添加,一样版本和默认路径的可直接复制粘贴):
在这里插入图片描述


CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1CUDA_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里添加变量,双击Path添加如下变量(同样根据实际路径填):

在这里插入图片描述

%CUDA_LIB_PATH%%CUDA_BIN_PATH%%CUDA_SDK_LIB_PATH%%CUDA_SDK_BIN_PATH%C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\lib\x64C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\bin C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\common\lib\x64C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.1\bin\win64

2.4检验是否安装成功
打开cmd命令窗口,切换路径(注意根据实际路径)

cd C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.1\extras\demo_suite

分别运行以下命令

bandwidthTest.exe
deviceQuery.exe

若分别显示Result = PASS,则安装成功,如图:
在这里插入图片描述
3.安装cuDNN
cuDNN是CUDA的库,易于写代码。官网下载需要先注册英伟达账号。
3.1下载
官网下载地址:点这里
注意和前面的表格里的版本要求相匹配,这里我下载的是7.6.5,如图:
在这里插入图片描述

百度网盘下载链接 提取码:ea1b

3.2解压下载好的软件
解压后进入文件夹,按照下面的要求将文件放到相应的位置:
将解压后文件夹中include、lib/x64和bin文件夹中的.h .lib 和.dll 后缀的文件分别拷贝到cuda对应路径下的include, lib/x64,bin 文件夹下。

到这里就安装好了CUDA10.0及其对应的cuDNN文件,艰难的路程已经基本走完。

4.安装Tensorflow-gpu
4.1创建环境
打开Anaconda Prompt
在这里插入图片描述
先在命令行使用以下命令查看conda版本

conda --version

如果显示的版本是4.6.11,建议使用以下命令更新conda

conda update conda

接着,在命令窗口输入以下命令创建环境:

conda create -n tf-gpu python=3.7

然后激活刚创建的环境

activate tf-gpu

4.2安装
查看可下载版本

conda search tensorflow-gpu

结果如图:
在这里插入图片描述
然后选择自己想要安装的版本,这里以2.1.0为例

conda install tensorflow-gpu=2.1.0

4.3测试安装是否成功
打开命令提示符,进入Anaconda3\envs\tf-gpu文件夹,输入python

注:tf-gpu是创建环境时的命名

然后输入以下两个测试程序 (一行一行的输入)

  • 程序1
import tensorflow as tf
tf.__version__

输出结果
在这里插入图片描述

  • 程序2
import tensorflow as tf
tf.test.is_gpu_available()

输出结果
在这里插入图片描述
最后结果是True即可

如果以上测试正确,说明安装成功

5.将Tensorflow-gpu环境导入到VS Code中
5.1安装VS Code的Python插件
在这里插入图片描述

5.2选择Python解释器

安装Anaconda和创建tensorflow后,Anaconda和Tensorflow环境下都会拥有Python解释器,应该选择Tensorflow下的解释器。

在这里插入图片描述
在这里插入图片描述

到此就可以大功告成了!!!


http://www.ppmy.cn/news/739855.html

相关文章

【CubeMX配置STM32并驱动0.96寸OLED】

CubeMX配置STM32并驱动0.96寸OLED 一、使用CubeMX进行基础的配置 1、配置时钟 这里的时钟我试了一下,设置为72MHZ和32MHZ好像对IIC的影响区别不是很大,这个就看个人吧。 1、配置IIC 在这里只需要选择需要用的IICX就可以了。因为我使用的是C8T6&#xf…

win11 使用wsl2安装pytorch 梳理

文章目录 前言一、wsl2 ubuntu xfce4 VcXsrv xrdp 远程桌面连接 及其改进 MobaXterm二、安装pytorch三. ubuntu命令行可视化anaconda总结 前言 本文章的内容是我在使用ubuntu安装pytorch的过程,主要的关键词是 win11 - wsl2 - MobaXterm - anaconda3 - pytorc…

完美解决TensorFlow-gpu报错问题Could not load dynamic library ‘libnvinfer.so.6’ and ‘libcudart.so.11.0’

一、问题描述 Could not load dynamic library ‘libnvinfer.so.6’; dlerror: libnvinfer.so.6: cannot open shared object file: No such file or directory; LD_LIBRARY_PATH: :/usr/local/cuda-10.1/lib64 报错后仍可以正常运行,但是发现GPU并未调用&#xff0…

Internal: no kernel image is available for execution on the device Fatal Python error: Aborted

使用tensorflow2.3.1训练模型时,遇到了下面的错误: StreamExecutor device (0): NVIDIA GeForce MX130, Compute Capability 5.0 2022-02-13 21:03:08.071293: F .\tensorflow/core/kernels/random_op_gpu.h:232] Non-OK-status: GpuLaunchKernel(FillP…

win10 电脑自带GPU Pytorch却无法通过CUDA使用

博主遇到的问题: 笔记本自带NVIDIA GeForce MX130独立显卡,但torch.cuda.is_available()却返回False。通过查找资料,总结解决流程如下: 解决流程 启动NVIDIA相关服务 参考教程中的方法一:https://www.xitongtiandi.…

调用GPU不匹配“cuda:0“

👦👦一个帅气的boy,你可以叫我Love And Program 🖱 ⌨个人主页:Love And Program的个人主页 💖💖如果对你有帮助的话希望三连💨💨支持一下博主 GPU不匹配"cuda:0&…

cuda10.1+cudnn10.1+tensorflow2.2.0+pytorch1.7.1下载安装及配置

一、cuda及cudnn下载 1、查看自己电脑是否支持GPU 方法:鼠标移动到此电脑,点击鼠标右键,依次选择属性、设备管理器、显示适配器有以下图标(NVIDIA)即可安装GPU,我的是MX130的。 2、选择和自己电脑相匹配的…

【CubeMX配置STM32驱动MPU6050】

CubeMX配置STM32驱动MPU6050(包含DMP)并且在0.96寸OLED上显示 一、使用CubeMX进行相关配置 1、配置OLED的IIC接口 OLED的具体使用方法我就不细说了,我前面的文章里面有讲OLED的,如果有需要可以去看看。 2、配置MPU6050的接口 我…