Differences

This shows you the differences between two versions of the page.

--- tutoriais:conda [2024-04-16 18:49] – gnann
+++ tutoriais:conda [2024-04-16 19:33] (current) – [Torch] gnann
@@ Line 25: / Line 25: @@
 ===== Tensorflow =====
 <code bash>
-# python=3.9
+# DISCLAIMER: versão 2.4 !
-conda install tensorflow-gpu
+conda install tensorflow-gpu numpy=1.23
+</code>
+<code python>
+import tensorflow as tf
+cuda_version = tf.sysconfig.get_build_info()["cuda_version"]
+gpu = tf.test.gpu_device_name()
+print("GPU: "+gpu)
+print("cuda: "+cuda_version)
 </code>
@@ Line 32: / Line 41: @@
 <code bash>
 # python do sistema
-conda install pytorch-gpu
+conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
 </code>
+<code python>
+import torch
+print("gpu: "+torch.cuda.get_device_name(0))
+print("cuda: "+torch.version.cuda)
+</code>
+https://pytorch.org/get-started/locally/