凛冬将至

从简单的例子开始

0%

win10配置GPU版TensorFlow.md

尝试在win10的台式机上,把我的3060用起来。

安装Anaconda

我已经安装好了anaconda,但还是需要配置环境变量

1
2
D:\ProgramData\Anaconda3\Scripts
D:\ProgramData\Anaconda3\pkgs\python-3.8.8-hdbf39b2_5

确认GPU类型

查看GPU类型

然后用谷歌搜索“NVIDIA GeForce RTX 3060+SPECIFICATION”,翻到最后在官网查它是否支持CUDA,可以看到其支持的列表,我的既然写了CUDA cores数量应该是支持的。

TensorFlow官网查询版本和python版本、CUDA等对于关系。

安装CUDA

下载

下载CUDA,去官网下载并傻瓜式安装。我这里下载的是CUDA Toolkit 11.0

安装

尝试了一下,在cmd中输入nvcc -V,显示cuda版本为V11,安装成功。

安装cuDNN

依然是去官网下载,下载之后解压,并将文件夹下的文件拷贝到cuda中与之相对应的文件夹下,即可。

安装tensorflow

直接安装tensorflow2.5版即可

1
pip install tensorflow==2.5

安装需要的库

1
2
conda install matplotlib
conda install pandas

测试是否成功

使用代码测试是否可以调用GPU,运行以下代码。

1
2
3
4
5
6
7
8
9
import tensorflow as tf
import os
tf.config.list_physical_devices('GPU')

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

a = tf.constant(2.0)
b = tf.constant(4.0)
print(a + b)

并没有成功,因为很多dll文件,tensorflow找不到,是环境变量问题。根据报错提示,去官网看了一下。让我将这些路径加到PATH中。其中cuDNN需要解压后拷贝一下,我这里按他的提示,拷贝到C:\tools\cuda,并增加到系统路径中。

都加了之后确实大部分dll文件都能找到了,但下面这个cusolver64_11.dll还是找不到,去cuda的文件夹(C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.0)中搜了一下,发现有个cusolver64_10.dll,但没有cusolver64_11.dll。于是我直接粗暴地将这个文件复制一下,新文件命名为cusolver64_11.dll,再测试一下,成功了。