Tensorflow和pytorch判断GPU是否可用

2022-10-05 11:57:20

首先进入python编译环境

Pytorch

import torch
torch.cuda.is_available()

显示true则GPU可用

Tensorflow

import tensorflowas tf
tf.test.is_gpu_available()

显示true 则GPU可用

附：

  1. 查看cuda版本
nvcc -V
  1. 查看torch/tensorflow版本
import torch/import tensorflowas tf 
torch/tf/__version__
  • 作者：wanghx7
  • 原文链接：https://blog.csdn.net/wanghx7/article/details/119328532
    更新时间：2022-10-05 11:57:20