apply latest cuda nvcc and tensorrt to support 50 serious gpu
This commit is contained in:
1 parent
c413e4f32f
commit
7183223be3
2 files changed
+4
-4
No files matched your search
+2
-2
@@ -28,13 +28,13 @@ call conda create -y -n ezvtb_rt_venv python=3.10
|
||||
call conda activate ezvtb_rt_venv
|
||||
call conda env list
|
||||
|
||||
call conda install -y nvidia/label/cuda-12.6.3::cuda-nvcc-dev_win-64
|
||||
call conda install -y nvidia/label/cuda-12.9.0::cuda-nvcc-dev_win-64
|
||||
call conda install -y conda-forge::pycuda
|
||||
|
||||
call python -m pip install --upgrade pip wheel
|
||||
echo yes|python -m pip install nvidia-cudnn-cu12
|
||||
|
||||
echo yes|pip install tensorrt_cu12_libs==10.6.0 tensorrt_cu12_bindings==10.6.0 tensorrt==10.6.0 --extra-index-url https://pypi.nvidia.com
|
||||
echo yes|pip install tensorrt_cu12_libs==10.11.0.33 tensorrt_cu12_bindings==10.11.0.33 tensorrt==10.11.0.33 --extra-index-url https://pypi.nvidia.com
|
||||
|
||||
call python -m pip install -r requirements.txt --no-warn-script-location
|
||||
|
||||
|
||||
+2
-2
@@ -28,13 +28,13 @@ call conda create -y -n ezvtb_rt_venv python=3.10
|
||||
call conda activate ezvtb_rt_venv
|
||||
call conda env list
|
||||
|
||||
call conda install -y nvidia/label/cuda-12.6.3::cuda-nvcc-dev_win-64
|
||||
call conda install -y nvidia/label/cuda-12.9.0::cuda-nvcc-dev_win-64
|
||||
call conda install -y pycuda -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/
|
||||
|
||||
call python -m pip install --upgrade pip wheel -i https://mirrors.aliyun.com/pypi/simple/
|
||||
echo yes|python -m pip install nvidia-cudnn-cu12 -i https://mirrors.aliyun.com/pypi/simple/
|
||||
|
||||
echo yes|pip install tensorrt_cu12_libs==10.6.0 tensorrt_cu12_bindings==10.6.0 tensorrt==10.6.0 --extra-index-url https://pypi.nvidia.com
|
||||
echo yes|pip install tensorrt_cu12_libs==10.11.0.33 tensorrt_cu12_bindings==10.11.0.33 tensorrt==10.11.0.33 --extra-index-url https://pypi.nvidia.com
|
||||
|
||||
call python -m pip install -r requirements.txt --no-warn-script-location -i https://mirrors.aliyun.com/pypi/simple/
|
||||
|
||||
|
||||
Reference in new issue
Block a user