欢迎光临散文网 会员登陆 & 注册

Win10系统1050ti显卡安装cuda11.5和cudnn8.4,TensorFlow2.6.0

2022-04-10 00:42 作者:M张小晨  | 我要投稿

本文只是针对1050ti显卡在2022年4月9日的安装方法,采用python3.7,cuda11.5,cudnn8.4,vs2022。本文最终只支持GPU True tf.Tensor(6.0, shape=(), dtype=float32)的结果,Anaconda和pycharm的安装以及具体的细节强烈建议先看这位up的视频https://www.bilibili.com/video/BV1Lv41177BW

 

测试TensorFlow安装的结果


1.确定安装版本

在桌面用win+R输入cmd进入控制台,输入nvidia-smi

或者在桌面右键进入英伟达控制面板

看到图中支持的版本是cuda11.5,接着在TensorFlow官网 https://tensorflow.google.cn/install/source_windows查看对应的cudnn版本。


目前还没有找到11.5,所以我选择了python3.7、cudnn在官网找的8.4版本,vs之前用的时候已经装了2022版也没有卸载。

2. 安装cuda11.5和cudnn8.4

在官网https://developer.nvidia.com/cuda-toolkit-archive选择cuda11.5


在cudnn官网NVIDIA cuDNN | NVIDIA Developerhttps://developer.nvidia.com/cudnn选择8.4下载



至此,针对1050ti的两个文件为:




双击cuda的exe先解压,然后安装,然后把第二个压缩包的三个文件放到cuda的安装目录里面粘贴替换。

上图为cuda的安装目录,将压缩包的是三个文件夹替换进去


在控制台输入nvcc -V查看是否安装成功


 

3.创建conda虚拟环境(具体细节请参考视频)

  1. 在桌面用win+R输入cmd进入控制台;

  2. 控制台输入conda create -n tf26 python==3.7,输入y,创建完成;

  3. 输入activate进入base环境;

  4. 输入conda activate tf26进入虚拟环境,环境名称就是tf26,可以任意;

输入

pip install tensorflow_gpu==2.6.0 -i https://pypi.douban.com/simple --trusted-host pypi.douban

或者

pip install tensorflow_gpu==2.6.0 -i https://pypi.douban.com/simple

即可进行安装TensorFlow2.6。文件大概340Mb.

 

测试是否安装成功可以参见https://blog.csdn.net/qq_35456045/article/details/104666083

4.在pycharm中添加python解释器

在pycharm中依次点击:文件-设置-python解释器-显示所有-添加之前创建的虚拟环境tf26-确定(如下图所示),需要等pycharm扫描这个解释器,等右下角进度条跑完,强烈建议先重启pycharm



然后输入代码。

import tensorflow as tf

 

#查看tensorflow版本

print(tf.__version__)

 

print('GPU', tf.test.is_gpu_available())

 

a = tf.constant(2.0)

b = tf.constant(4.0)

print(a + b)

 


如图,显示gpu false,计算结果为2+4=6.

 如果遇到cudart64_110.dll 文件不存在,下载后放到c\windows\system32下面https://www.dll-files.com/cudart64_101.dll.html

 

 

 

主要参考以下资料:

 

  1. 【包教包会】Anaconda+Cuda+Cudnn+TensorFlow+Pytorch+Pycharm+Win10深度学习环境配置与卸载https://www.bilibili.com/video/BV1Lv41177BW

  2. Install Tensorflow GPU on Windows https://www.youtube.com/watch?v=EmZZsy7Ym-4

  3. error: Ignore above cudart dlerror if you do not have a GPU set up on your machine. https://blog.csdn.net/qq_43355165/article/details/121879462

  4. TensorFlow官网 https://tensorflow.google.cn/install/source_windows

  5. 安装cuda和cudnn https://blog.csdn.net/linxinloningg/article/details/122522975

  6. TensorFlow2.x GPU版安装与CUDA版本选择指南 https://www.csdn.net/tags/MtjaAg5sMDk4ODQtYmxvZwO0O0OO0O0O.html

  7. cudart64_110.dll not found windows解决方法 https://blog.csdn.net/weixin_42474121/article/details/115708919

  8. Win10+1050Ti配置tensorflow-gpu教程 (解决1050ti配置cuda失败的问题)https://blog.csdn.net/mch2869253130/article/details/88889616

  9. win10 1050Ti 笔记本配置 TensorFlow-gpu 过程(多图超详细)https://blog.csdn.net/weixin_43626465/article/details/115706195

  10. AttributeError: module 'tensorflow' has no attribute 'Session'错误解决 https://blog.csdn.net/sinat_36502563/article/details/102302392

 最后,本文只是针对1050ti显卡在2022年4月9日的安装,采用python3.7,cuda11.5,cudnn8.4,vs2022。本文最终只支持GPU True tf.Tensor(6.0, shape=(), dtype=float32)的结果,不过后来听说测试结果为GPU false 其实也不影响运行,其他代码还没有测试。


Win10系统1050ti显卡安装cuda11.5和cudnn8.4,TensorFlow2.6.0的评论 (共 条)

分享到微博请遵守国家法律