conda create -n tf21 python=3.7
conda activate tf21
conda install tensorflow-gpu==2.1

验证

import tensorflow as tf
tf.test.is_built_with_cuda()

在这里插入图片描述

Logo

汇聚全球AI编程工具,助力开发者即刻编程。

更多推荐