cuda10.2安装tensorflow 1.14.0

客官°小女子只卖身不卖艺 2022-09-09 00:08 233阅读 0赞

# 背景 #

最近为了跑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][win10_cuda10.2_CUDNN8.2.2]

# 重新安装tensorflow-GPU步骤 #

### 创建环境 ###

conda create --name tensorflow_gpu python=3.6

### 激活环境 ###

activate tensorflow_gpu

### 查看环境 ###

conda info --envs

![在这里插入图片描述][08ee25b668924775b3583e7fc846d208.png]

### 安装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增加一个环境 #

目前有了两个环境，增加一个刚才新建那个，如下图  
![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA57-f57695ZqE_size_20_color_FFFFFF_t_70_g_se_x_16]

# 测试成功 #

![在这里插入图片描述][watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA57-f57695ZqE_size_20_color_FFFFFF_t_70_g_se_x_16 1]

[win10_cuda10.2_CUDNN8.2.2]: https://blog.csdn.net/mao_hui_fei/article/details/119564307
[08ee25b668924775b3583e7fc846d208.png]: /images/20220829/2ee3246018014d73804012a9a9991ce0.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA57-f57695ZqE_size_20_color_FFFFFF_t_70_g_se_x_16]: /images/20220829/9e1391699a5246d88b6bf8277885225c.png
[watermark_type_ZHJvaWRzYW5zZmFsbGJhY2s_shadow_50_text_Q1NETiBA57-f57695ZqE_size_20_color_FFFFFF_t_70_g_se_x_16 1]: /images/20220829/514ee38e2d5445afa9a837bc17b51bc1.png