终于用我的那个二手显卡搭建好了 TF2 的环境
上周,心血来潮在咸鱼淘了一张 GTX1060 3G 的“丐中丐”的显卡。打算来搭建一下 TF2 的 GPU 环境,从此我开始了我噩梦般的周末
环境介绍
OS:WIN10
Python:3.8.3
GPU:GTX1060 3G
CUDA:11.1
CUDNN:cudnn-11.2-windows-x64-v8.1.0.77
有的书上说要输入这个命令来按照 GPU 版本:
复制代码
现在 TF 的最新版本默认的就是 CPU+GPU 的版本,因此不用特地指定。
我遇到的问题:
conda 安装了识别不到,然后尝试 pip 安装,运行时报各种 dll 找不到。折腾了好久,各种版本的 CUDA 和 cudnn 来回换。
解决办法
基本所有的 dll 在 CUDA 的 bin 目录下都有,将缺失的复制到 C:\Windows\System32,然后就可以正常运行了,如果某个 dll 找不到,可以上网搜一下。
我已经试过添加 path,但是没有效果,才选择了这样比较麻烦的方式。
评论