• Tensorflow/Pytorch with GPU on ARM64

    Maciek
    Maciek

    Hi,

    I wish to launch some ML python scripts on arm64 and I have a question how one could use tensorflow or pytorch with gpu acceleration using nvidia drivers. As I can see, for ARM64 NVIDIA supports only CUDA 11. However tensorflow is only compatible…

    • 2 months ago
    • Open Source Software and Platforms
    • Machine Learning forum
  • View related content throughout Open Source Software and Platforms
  • More
  • Cancel