Skip to content

Commit

Permalink
Update Installer for Windows. Open for reading in the editor.bat
Browse files Browse the repository at this point in the history
  • Loading branch information
My12123 authored Dec 23, 2024
1 parent 32cb59a commit 8c56fe0
Showing 1 changed file with 5 additions and 3 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -3,12 +3,14 @@
@rem Install python 3.10
@rem Pre-installation steps copy files from TensorRT-8.6.0.12\onnx_graphsurgeon and TensorRT-8.6.0.12\python and place in .\
@rem This installer supports TensorRT-8.6.0.12 for now but you can change that to whatever version you need.
@rem Post-Installation Steps: Download and copy files from C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8\bin and TensorRT\lib to .\venv\Scripts\
@rem Post-Installation Steps: Download and copy files from TensorRT\lib to .\venv\Scripts\
@rem This is necessary so as not to add to PATH.
python -m venv venv
call .\venv\Scripts\activate
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu121
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu124
pip install -r requirements.txt
pip install onnx_graphsurgeon-0.3.12-py2.py3-none-any.whl
pip install tensorrt-8.6.0-cp310-none-win_amd64.whl
call pip install git+https://github.com/NVIDIA-AI-IOT/torch2trt.git
call git clone https://github.com/NVIDIA-AI-IOT/torch2trt.git
call cd torch2trt
call python setup.py install

0 comments on commit 8c56fe0

Please sign in to comment.