如何在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

腾讯云面向开发者汇聚海量精品云计算使用和开发经验,营造开放的云计算技术生态圈。

更多推荐