search

reid41 / QA-Pilot

QA-Pilot is an interactive chat project that leverages online/local LLM for rapid understanding and navigation of GitHub code repository.
Apache License 2.0
165 stars 17 forks source link

Cannot install torch==2.2.2+cu121 on Ubuntu 22.04 / conda #11

Open vsigal opened 5 months ago

vsigal commented 5 months ago

Got : ERROR: Could not find a version that satisfies the requirement torch==2.2.2+cu121

Solved it by executing command :
conda install pytorch==2.2.2 torchvision==0.17.2 torchaudio==2.2.2 pytorch-cuda=11.8 -c pytorch -c nvidia

and removing three lines in file requirements.txt : torch==2.2.2+cu121 torchaudio==2.2.2+cu121 torchvision==0.17.2+cu121

reid41 commented 5 months ago

thanks for testing, seems thecuda cannot directly use for the dependencies.. I will remove the cuda pkg in requirements.txt