...
| Code Block | ||||
|---|---|---|---|---|
| ||||
$ python3 --version Python 3.1013.1213 |
| Code Block | ||||
|---|---|---|---|---|
| ||||
$ nvcc --version nvcc: NVIDIA (R) Cuda compiler driver Copyright (c) 2005-2025 NVIDIA Corporation Built on FriWed_FebAug_2120_2001:2358:5059_PM_PSTPDT_2025 Cuda compilation tools, release 1213.80, V12V13.80.9388 Build cuda_1213.80.r12r13.80/compiler.3558387036424714_0 |
| Code Block | ||||
|---|---|---|---|---|
| ||||
# Check with conda which versions are installed $ conda list | grep -E 'tensorflow|keras' keras 3.14.1 pyh753f3f9_0 conda-forge libtensorflow_cc 2.19.1 cuda128h5d964f1_255 conda-forge libtensorflow_framework 2.19.1 cuda128hc512d3a_255 conda-forge tensorflow 2.19.1 cuda128py312h8a47763_255 conda-forge tensorflow-base 2.19.1 cuda128py312ha24b813_255 conda-forge tensorflow-gpu 2.19.1 cuda128h780c90d_255 conda-forge |
...
| Code Block | ||||
|---|---|---|---|---|
| ||||
>>> import torch >>> torch.__version__ '2.1112.0+cu128cu130' >>> torch.cuda.is_available() True >>> torch.version.cuda '1213.80' # Create a tensor and move it to GPU >>> x = torch.tensor([1.0, 2.0]).cuda() tensor([1., 2.], device='cuda:0') |