Win10系统1050ti显卡安装cuda11.5和cudnn8.4,TensorFlow2.6.0
本文只是针对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。

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的安装目录里面粘贴替换。

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

3.创建conda虚拟环境(具体细节请参考视频)
在桌面用win+R输入cmd进入控制台;
控制台输入conda create -n tf26 python==3.7,输入y,创建完成;
输入activate进入base环境;
输入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
主要参考以下资料:
【包教包会】Anaconda+Cuda+Cudnn+TensorFlow+Pytorch+Pycharm+Win10深度学习环境配置与卸载https://www.bilibili.com/video/BV1Lv41177BW
Install Tensorflow GPU on Windows https://www.youtube.com/watch?v=EmZZsy7Ym-4
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
TensorFlow官网 https://tensorflow.google.cn/install/source_windows
安装cuda和cudnn https://blog.csdn.net/linxinloningg/article/details/122522975
TensorFlow2.x GPU版安装与CUDA版本选择指南 https://www.csdn.net/tags/MtjaAg5sMDk4ODQtYmxvZwO0O0OO0O0O.html
cudart64_110.dll not found windows解决方法 https://blog.csdn.net/weixin_42474121/article/details/115708919
Win10+1050Ti配置tensorflow-gpu教程 (解决1050ti配置cuda失败的问题)https://blog.csdn.net/mch2869253130/article/details/88889616
win10 1050Ti 笔记本配置 TensorFlow-gpu 过程(多图超详细)https://blog.csdn.net/weixin_43626465/article/details/115706195
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 其实也不影响运行,其他代码还没有测试。