July 2018
Intermediate to advanced
254 pages
8h 12m
English
We are now ready to install PyTorch! Fortunately, it is as simple as running the following command inside our rl_gym_bookconda environment:
(rl_gym_book) praveen@ubuntu:~$ conda install pytorch torchvision -c pytorch
Note that this command will install PyTorch with CUDA 8.0. You noted the CUDA version that you installed before, and the command may change slightly depending on which CUDA version you installed. For example, if you installed CUDA 9.1, the command to install will be this:
conda install pytorch torchvision cuda91 -c pytorch
You can find the updated command to install at http://pytorch.org based on your OS, package manager (conda or pip or from source), Python version (we use 3.5), and CUDA version.
That's ...
Read now
Unlock full access