os.environ["CUDA_VISIBLE_DEVICES"] = "" import tensorflow as tf from tacotron.models import create_model from tacotron_hparams import hparams import shutil #with tf.device('/cpu:0'): inputs = tf.placeholder(tf.int32, [1, None], 'inputs') input_lengths = tf.placeholder(tf.int32,...
os.environ['CUDA_VISIBLE_DEVICES'] = '0' import os import unittest from swift.llm import (export_main, ExportArguments) class TestTemplate(unittest.TestCase): def setUp(self): print(('Testing %s.%s' % (type(self).__name__, self._testMethodName))) self.tmp_dir = tempfile.TemporaryDi...
importos os.environ["CUDA_VISIBLE_DEVICES"]="" importtensorflowastf fromtacotron.modelsimportcreate_model fromtacotron_hparamsimporthparams importshutil #with tf.device('/cpu:0'): inputs=tf.placeholder(tf.int32,[1,None],'inputs') input_lengths=tf.placeholder(tf.int32,[1],'input_lengths')...
# devices: 1Device 0Name: NVIDIA GeForce RTX 2060Preferred: TRUEPower Envelope: DISCRETEAttachment: UNKNOWN# attached displays: 0GPU accessible RAM: 6,442 MBVRAM: 6,442 MBDedicated System RAM: 0 MBShared System RAM: 0 MBAPI version: 3.0 (OpenCL 3.0 CUDA)...
If you cannot interact, then the Save Dialog is probably open but not visible. If this is the case, it made have been moved to another location like the bottom of the screen or to a different monitor. You can test this by click the Enter or Esc key after clickin...
if get_model_parameter_size(model) > total_mem_per_cpu * 0.45: device_collection = [torch.device(i) for i in range(torch.cuda.device_count())] if len(device_collection) > 1: print( f"{len(device_collection)} GPUs are used to export onnx, \ Please set CUDA_VISIBLE_DEVICES to us...
import os #os.environ["CUDA_VISIBLE_DEVICES"] = "-1" import tensorflow as tf from tensorflow import keras from tensorflow.keras import layers from tensorflow.keras.applications import DenseNet121, MobileNetV3Small img_inputs = keras.Input(shape=(128, 128, 3)) backend = MobileNetV3Small(include...
If you cannot interact, then the Save Dialog is probably open but not visible. If this is the case, it made have been moved to another location like the bottom of the screen or to a different monitor. You can test this by click the Enter or Esc key after clicking on ...
If you cannot interact, then the Save Dialog is probably open but not visible. If this is the case, it made have been moved to another location like the bottom of the screen or to a different monitor. You can test this by click the Enter or Esc key after c...
Cleanup torch amp usage to avoid cuda specific calls, merge support for Ascend (NPU) devices from MengqingCao that should work now in PyTorch 2.5 w/ new device extension autoloading feature. Tested Intel Arc (XPU) in Pytorch 2.5 too and it (mostly) worked. Oct 16, 2024 Fix error on im...