You can install any available Python packages you need with either pip and/or conda. However, due to technical reasons the default environment will always reset between sessions, so we need take extra care to ensure that packages are installed on your own persistent storage instead of the default environment.
...
If initializing conda worked correctly, we can then create a new conda environment. With the '--prefix' flag we 're are creating an environment by specifying a location for instead of a name - named environments will get removed between sessions. You can change the location '~/myenv' to anything you might want.
...
Then we install the kernel and the new packages we want to use. For this simple example we're only installing 'pytorch'. You must always make sure to include the 'ipykernel' package in the environment, no matter which other packages you want to install!
...