天坑!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 测试
-
pytorch gpu 测试
参考: