admin管理员组

文章数量:1655036

 

1)安装好anaconda3并升级到最新版本

2)在NVIDIA官网上下载最新的GTX1650网卡的驱动,进行重新安装。

3)在CUDA下载地址:https://developer.nvidia/cuda-toolkit-archive,本机用的:CUDAcuda_10.0.130_411.31_win10

并进行默认安装(默认存在C盘,其它盘没有尝试)。系统会自动添加环境变量。

4)CuDNN下载地址:https://developer.nvidia/rdp/cudnn-download(该网站需要注册),下载对应的CuDNN10.0,下载好之后,解压,分别将.h .lib 和.dll 文件分别拷贝到cuda的include, lib/x64, bin 文件夹下。其实也就是对应的文件夹。

5)环境变量配置(参考:https://blog.csdn/qq_33287871/article/details/101112357):

还需要自己添加几个系统变量
CUDA_SDK_PATH = C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.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里添加东西。找到系统变量的PATH双击,添加%CUDA_LIB_PATH%
%CUDA_BIN_PATH%
%CUDA_SDK_LIB_PATH%
%CUDA_SDK_BIN_PATH%
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\lib\x64 
C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\bin
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\common\lib\x64
C:\ProgramData\NVIDIA Corporation\CUDA Samples\v10.0\bin\win64
(有些系统会自动添加,如果PATH里已经有了就不用添加了)

6)添加完之后CUDA就算安装完成了。检验是否安装成功,打开CMD(1)输入nvcc -V,查看版本是不是10.0,(2)cd到C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v10.0\extras\demo_suite (这是默认路径)
分别执行命令:bandwidthTest.exe;deviceQuery.exe

 7)安装TensorFlow环境:conda create -n tensorflow2 python=3.7,安装成功后激活,activate tensorflow2,进入TensorFlow2环境。

8)安装TensorFlow-GPU:pip isntall tensorflow-gpu==2.3.0(或者conda install tensorflow-gpu==2.3.0),等待安装成功。

9)重启电脑(我在安装时,没有重启电脑,出现.dll文件无法加载情况,重启后就没了。没有验证重新安装是否要重启电脑)

10)(1)测试GPU是否工作,在Python下(不知哪位大侠的代码,找不到地址了)

 import tensorflow as tf

print(tf.test.is_gpu_available())

如果是TRUE安装成功。

(2)测试代码:

 在命令行中,进入python,并输入以下代码:
import tensorflow as tf 

print(tf.__version__)
输出:
2.3.0
————————————————————————————————————————————————————————

搞了两天,除了环境变量配置,建议采用anaconda的虚拟环境,在虚拟环境下:“conda install tensorflow-gpu==2.3.0”一句命令,搞定所有。

本文标签: 专业版