Differences

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

--- tutoriais:conda [2024-04-16 19:01] – [Tensorflow] gnann
+++ tutoriais:conda [2024-04-16 19:33] (current) – [Torch] gnann
@@ Line 23: / Line 23: @@
 </code>
+===== Tensorflow =====
+<code bash>
+# DISCLAIMER: versão 2.4 !
+conda install tensorflow-gpu numpy=1.23
+</code>
+<code python>
 import tensorflow as tf
 cuda_version = tf.sysconfig.get_build_info()["cuda_version"]
@@ Line 29: / Line 36: @@
 print("GPU: "+gpu)
 print("cuda: "+cuda_version)
+</code>
 ===== Torch =====
 <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/