admin管理员组

文章数量:1646237

背景

最近为了跑yolov5,我要安装pytorch,然后将cuda 10.0升级到了cuda10.2.pytorch是没问题了,但是我再运行我的yolo v4(tensorflow框架)代码时发现运行不了了,提示:cudart64_100.dll找不到,但是我现在是cuda 10.2了,他还在找cuda10.0。

然后网上各种说cuda 10.2没有对应的tensorflow版本,我就觉得再装回去太麻烦了。下面是我的解决办法。

win10安装cuda10.2和CUDNN8.2.2

见我另外一篇博客:win10安装cuda10.2和CUDNN8.2.2

重新安装tensorflow-GPU步骤

创建环境

conda create --name tensorflow_gpu python=3.6

激活环境

activate tensorflow_gpu

查看环境

conda info --envs

安装tensorflow

conda install tensorflow-gpu==1.14.0

安装keras

pip install keras==2.2.5

安装其他库

pip install Pillow
pip install matplotlib
pip install opencv-python
pip install xlwt
pip install PyQt5

Pycharm增加一个环境

目前有了两个环境,增加一个刚才新建那个,如下图

测试成功

本文标签: tensorflow