User Tools

Site Tools


compute:pythongpu

Differences

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

Link to this comparison view

Next revision
Previous revision
compute:pythongpu [2024/12/11 15:16] – created bjornscompute:pythongpu [2024/12/11 15:48] (current) bjorns
Line 4: Line 4:
 Both Tensorflow and Torch have been used on this cluster succesfully.\\ Both Tensorflow and Torch have been used on this cluster succesfully.\\
 \\ \\
-An example setup using Python venv, run these commands\\+An example setup using Python venv, run these commands :\\ 
 +\\
 ''python3 -m venv myTorch''\\ ''python3 -m venv myTorch''\\
 ''source myTorch/bin/activate''\\ ''source myTorch/bin/activate''\\
 ''pip install torch''\\ ''pip install torch''\\
-The new Python virtual environment has been created and activated, torch package installed\\ +\\ 
-And the correct CUDA library is also installed\\+The new Python virtual environment has been created and activated, **torch** package installed\\ 
 +And the correct **CUDA library** is also installed\\
 Please run a simple test with python code to make sure you can communicate with the GPU\\ Please run a simple test with python code to make sure you can communicate with the GPU\\
 +One example command to test torch\\ 
 +''if torch.cuda.is_available() print("GPU")'' 
 +\\ 
 +But you must run this code test with **sbatch**, send your job to the cluster\\ 
 +Create a python file dummy.py with the test-GPU command\\ 
 +\\ 
 +Update the sbatch script file, mine is called **oneJob.sh**\\ 
 +Remove lines below last #SBatch\\ 
 +Add 2 lines\\ 
 +''source myTorch/bin/activate''\\ 
 +''python3 dummy.py''\\ 
 +[SAVE]\\ 
 +Run the command ''sbatch oneJob.sh''\\ 
 +The response should be **Submitted batch job XXX**\\ 
 +Then check your log file (default name is myJob.log) for the result\\ 
 +\\ 
 +Installing Tensorflow is same process, we recommend using a different virtual environment if you want to try both of them\\ 
 +Make sure to instruct pip in this way, to get all necessary packages ''pip install "tensorflow[and-cuda]"''\\
 \\ \\
 == Those version have also been used before and known to work == == Those version have also been used before and known to work ==
compute/pythongpu.1733930178.txt.gz · Last modified: 2024/12/11 15:16 by bjorns