TensorFlow GPU版安装指南(Linux系统)
发布时间
阅读量:
阅读量
一、TensorFlow-gpu环境的安装
在进行该代码的安装操作之前,需确保系统中已配置好相关的深度学习运行环境
(例如:conda、pytorch、cuda、cudnn等软件环境)
TensorFlow的版本需与所使用的GPU版本相匹配,用户可根据自身需求自行选定相应版本,亦可不指定具体版本直接完成TensorFlow-gpu的安装

conda create -n tf3 python==3.8
conda activate tf3

当前所配置的CUDA运行环境版本为11.8,Python解释器版本为3.8,同时安装了tensorflow-gpu==2.7.0的软件包。
具体安装步骤如下:
pip install tensorflow-gpu==2.7.0 -i https://pypi.mirr
全部评论 (0)
还没有任何评论哟~
