![](https://dkamatblog.home.blog/wp-content/uploads/2020/12/f6dbd-1chieypfnfd6dvldr6p9_-q.png?w=487&h=151)
The purpose of this blog post is to outline the steps to create a new PyTorch virtual environment
Firstly, Download and install Anaconda (choose the latest Python version).
Login to Anaconda prompt and create a .yml file (pytorch_env.yml) with below configuration
name: pytorch_gpu channels: — defaults — pytorch dependencies: — numpy=1.16.2 — pandas=0.24.2 — matplotlib=3.0.3 — pillow=5.4.1 — pip=19.0 — plotly=3.7.0 — scikit-learn=0.20.3 — seaborn=0.9.0 — python=3.7.3 — jupyter=1.0.0
Next, create a virtual environment using the .yml file, like so
conda env create -f pytorch_env.yml Collecting package metadata (repodata.json): done Solving environment: done
==> WARNING: A newer version of conda exists. <== current version: 4.7.12 latest version: 4.8.3
Please update conda by running
$ conda update -n base -c defaults conda
Downloading and Extracting Packages Preparing transaction: done Verifying transaction: done Executing transaction: done
# # To activate this environment, use # # $ conda activate pytorch_gpu # # To deactivate an active environment, use # # $ conda deactivate
Activate the newly created conda environment
#Get the list of available conda environments (base)C:> conda env list # conda environments: # base * C:UsersdivyaAnaconda3 kaggle C:UsersdivyaAnaconda3envskaggle nlp C:UsersdivyaAnaconda3envsnlp pytorch_gpu C:UsersdivyaAnaconda3envspytorch_gpu
#Activate conda environment
(base)C:> conda activate pytorch_gpu (pytorch_gpu)C:>
Check CUDA Version using below command
C:Users>nvidia-smi
CUDA version on my Machine is - CUDA Version: 10.1
Then, go to PyTorch’s site and find the get started locally section.
Specify the appropriate configuration options for your particular environment.
![](https://dkamatblog.home.blog/wp-content/uploads/2020/12/09c32-1vead7wda_66aj1tz3aaesg.png?w=1026&h=415)
Run the presented command in the terminal to install PyTorch.
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch
Finally, verify the installation by invoking a jupyter notebook
![](https://dkamatblog.home.blog/wp-content/uploads/2020/12/3002c-12po9indehwimbksrnwv1oq.png?w=1384&h=714)
The primary intention of writing this was for my future reference, I hope this also helps others to save some time when they want to get started with PyTorch. Thankyou!