在云服务器上安装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'))
标签:智享,whl,服务器上安装,cp38,版本,tensorflow,gpu
From: https://blog.csdn.net/weixin_45674669/article/details/136913131