cgpotts / cs224u

Code for Stanford CS224u

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Course setup, Pytorch CPU

eyast opened this issue · comments

Hi - I've followed the instructions to setup an environment for the course on my machine. The only difference I am aware of is that miniconda was pre-installed.
When following the instructions, the version of pytorch installed was cpu only. To fix that, I ran the following command:

conda install pytorch=1.13.1 pytorch-cuda=11.7 -c pytorch -c nvidia
This seems to have resolved it for me.

Sounds good! The setup.ipynb notebook asks people to go to the PyTorch homepage to choose their preferred method of installing PyTorch. As far as I know, this is the best solution for now, as the best choice is highly dependent on people's goals and computers.