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 ...