One thought on “How to setup PyTorch Jupyter Notebook on Intel Nervana AI Cluster (Colfax) For Deep Learning”

  1. i’m having a isue with jupyter notebook.
    i did the steps above, and created the pytorch and all the other steps, but when i create a notebook on jupyter it show that error:

    ImportErrorTraceback (most recent call last)
    in ()
    1 # test import torch and run some codes
    —-> 2 import torch
    3 a = torch.randn(5, 7)
    4 print(a)
    5 print(a.size())

    ImportError: No module named ‘torch’

    can you help me?

Leave a reply