Can I run assignment3 locally? about cs231n

[mantasu]

Yes, you will need GPU acceleration which is supported by PyTorch for M1 chips as well. You can install PyTorch by following this guide https://developer.apple.com/metal/pytorch/. Then just install the requirements:

pip install -r assignment3/requirements.txt

[ge1mina023]

thanks~

[ge1mina023]

What should I do additionally? Only comment out the code about google colab?

[mantasu]

You should also do the following changes in the files that actually use GPU acceleration:

1. In Generative_Adversarial_Networks.ipynb, whenever .type(dtype) is called, change that to .to(device="mps")
2. In Self_Supervised_Learning.ipynb change the final line in the 4th code cell:

   # device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
   device = torch.device("mps" if torch.backends.mps.is_available() else "cpu")

[ge1mina023]

Is that all I need to do?

[mantasu]

I can't confirm since I don't have Apple M1. It should be all but if you encounter any issues let me know.

[ge1mina023]

I can't confirm since I don't have Apple M1. It should be all but if you encounter any issues let me know.

thanks a lot~