如何在Linux 环境下安装tensorflow的gpu版本,记录安装的过程

  • 1、首先,创建一个python 环境,这里命名为tf-g,python选用的版本为3.8
conda create -n tf-g python=3.8
  • 2、然后,激活环境,并安装tensorflow的gpu版本
conda activate tf-g
conda install -c anaconda tensorflow-gpu
  • 3、 安装segmentation-models,这里面提供了很大好用的loss函数和封装好的模型
pip install segmentation-models
  • 4、安装基本的库、绘图、地形、读取nc文件、colormap
conda install -c conda-forge matplotlib seaborn cartopy netcdf4 cmaps
  • 5、科学计算相关的库
 conda install -c conda-forge  sklearn-contrib-lightning
  • 6、安装图像处理界面 spyder
 conda install -c conda-forge matplotlib spyder
  • 7、 安装深度学习相关的库 keras的gpu版本
conda install -c anaconda keras-gpu
Logo

分享最新的 NVIDIA AI Software 资源以及活动/会议信息,精选收录AI相关技术内容,欢迎大家加入社区并参与讨论。

更多推荐