...
The AMD-GPU-specific version of TensorFlow, Tensorflow-rocm 2.9.1 is installed on all four AMD GPU servers. This version works with ROCm 5.1.3+. If you need to install your own version with pip, specify this version:
Code Block | ||
---|---|---|
| ||
pip install tensorflow-rocm==2.9.1 |
You may also need to adjust your LD_LIBRARY_PATH as follows:
Code Block |
---|
export LD_LIBRARY_PATH="/opt/rocm-5.1.3/hip/lib:$LD_LIBRARY_PATH" |
PyTorch
PyTorch is available as a Docker image on hfogcomp02 and livecomp02. To run it, first define this alias:
...