?tensorflow

import tensorflow as tftf.test.is_gpu_available() # 或者 sess = tf.Session(config=tf.ConfigProto(log_device_placement=True))

torch

import torch flag = torch.cuda.is_available() print(flag)gpunumber = 1 # Decide which device we want to run on device = torch.device("cuda:0" if (torch.cuda.is_available() and gpunumber > 0) else "cpu") print(device) print(torch.cuda.get_device_name(0)) print(torch.rand(3,3).cuda())

總結(jié)

以上是生活随笔為你收集整理的tensorflow 测试 cuda 是否安装成功，测试代码环境的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。