search

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)
https://ffcv.io
Apache License 2.0
2.82k stars 178 forks source link

torchvision version cannot be updated #143

Closed AmmaraRazzaq closed 2 years ago

AmmaraRazzaq commented 2 years ago

I have installed ffcv with python 3.8, pytorch 1.8.1 and cuda 10.1.
conda create -y -n ffcv python=3.8 cupy pkg-config compilers libjpeg-turbo opencv pytorch=1.8.1 torchvision cudatoolkit=10.1 numba -c pytorch -c conda-forge However it installs a very old version of torchvision 0.2.2. I need atleast 0.9.1. However it is not being updated and if I try to create an environment with this specified version, the environment doesn't get created.

AmmaraRazzaq commented 2 years ago

Okay it worked for me. I did installation in steps.

  1. created a new environment with python=3.8
  2. conda install pytorch=1.8.1 torchvision=0.9.1 cudatoolkit=10.1 -c pytorch
  3. conda install cupy pkg-config compilers libjpeg-turbo -c conda-forge
  4. pip install opencv-python (installing it with conda was causing issues)
  5. pip install ffcv
  6. pip install -r requirements.txt