Install Pytorch with virtualenv

README.md

Create requirements.txt file & add some required libraries to install via pip

opencv-python
numpy
Pillow
matplotlib
tqdm

Getting Started

(1) Check if pip is installed:

$ pip --version

#If `pip` is not installed, follow steps below:
$ cd ~
$ curl https://bootstrap.pypa.io/get-pip.py -o get-pip.py
$ python3 get-pip.py

(2) Install virtual environment first & then activate:

$ cd <project-directory>
$ python3 -m pip install --user virtualenv #Install virtualenv if not installed in your system
$ python3 -m virtualenv env #Create virtualenv for your project
$ source env/bin/activate #Activate virtualenv for linux/MacOS
$ env\Scripts\activate #Activate virtualenv for Windows

(3) Install torch via pip by running following command (If possible always install torch via stable wheel they provide: https://pytorch.org/):

$ pip install torch==1.5.0+cu101 torchvision==0.6.0+cu101 -f https://download.pytorch.org/whl/torch_stable.html

(4) Install all dependencies for your project from requirements.txt file:

$ pip install -r requirements.txt

Hi Jay, thanks for the post here.

Can you include the version of Python you used, and graphics card and graphics card drivers used ?