Loading

天坑!win10 下 conda 安装 TensorFlow2.3 + Pytorch1.7(无需在系统中安装CUDA环境)

天坑!win10 下 conda 安装 TensorFlow2.3 + Pytorch1.7(无需在系统中安装CUDA环境)

重要的话说在前面:直接使用 conda install tensorflow-gpu=2.3 安装的版本是不能识别 GPU的。

为什么不用最新版?答:不兼容,tf23和torch17是最新的win下两个可以互相兼容的。

正确方案如下:

  • 安装最新显卡驱动

  • 安装 conda 提供的 cuda环境(一起把 cudnn 也安装了,两者版本需匹配)
    conda install cudatoolkit=10.1 cudnn=7.6.5

  • 安装 pytorch
    conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 -c pytorch

  • 安装 tf2(自动构建的 mkl 的 tf 包有问题,需要指定版本
    Python 3.7: conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py37h936c3e2_0
    Python 3.8: conda install tensorflow-gpu=2.3 tensorflow=2.3=mkl_py38h1fcfbd6_0

  • tf gpu 测试
    image

  • pytorch gpu 测试
    image

参考:

posted @ 2021-04-29 16:54  konosubaakua  阅读(349)  评论(0编辑  收藏  举报