背景
最近为了跑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增加一个环境
目前有了两个环境,增加一个刚才新建那个,如下图
测试成功