【恒源智享云】在云服务器上安装tensorflow_gpu教程

在云服务器上安装tensorflow_gpu

1.创建实例

在恒源智享云中创建新的实例时，可以选择需要使用的框架，并且框架版本和cuda版本、python版本都是对应的：

2.新建一个conda虚拟环境

conda create -n tensorflow_gpu python==3.8.0

conda指令详见博客：【恒源智享云】conda虚拟环境的操作指令

3.检查cuda是否安装

nvcc --version

如果显示版本号了，那就是已经安装了。一般，创建好实例后，恒源智享云已经配置好cuda、anaconda、python了。

4.下载tensorflow_gpu对应版本的whl文件

whl文件下载网址详见博客：【tensorflow_gpu】安装合集
whl文件的选择需要对应tensorflow版本、python版本，且因为是在服务器上运行，所以选择linux版本的。
根据上面创建实例的版本，下载的whl文件对应的应该是：

tensorflow_gpu-2.5.0-cp38-cp38-manylinux2010_x86_64.whl

5.将下载好的whl文件上传到服务器上

6.使用whl文件安装tensorflow_gpu

使用cd命令进入whl文件所在目录，运行以下指令安装tensorflow_gpu：

pip install tensorflow_gpu-2.5.0-cp38-cp38-manylinux2010_x86_64.whl

测试是否安装成功

# 测试tensorflow_gpu版本
import tensorflow as tf
print(tf.config.list_physical_devices('GPU'))