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