尝试在win10的台式机上,把我的3060用起来。
安装Anaconda
我已经安装好了anaconda,但还是需要配置环境变量
1 | D:\ProgramData\Anaconda3\Scripts |
确认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 | conda install matplotlib |
测试是否成功
使用代码测试是否可以调用GPU,运行以下代码。
1 | import tensorflow as tf |
并没有成功,因为很多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
,再测试一下,成功了。