On which CUDA version is the code run on?

[YUEFANG0211]

Dear author, thank u for your wonderful contribution!! It happens to me when I tried to reproduce your code.
It seems that torch2.0.1 is too old for cuda12. However, the Nvidia related packages are set to cu12. Why?
Thx!!

[sunnychenxiwang]

our cuda version is 12.4

[zxlzr]

hi, do you have any further questions?

[YUEFANG0211]

hi, do you have any further questions?

Thanks you for your reply!! I am still struggling with the environment setting😭 My current setting is as follows(CUDA18,python310). It doesn't work well. Thanks again and have a good day :)

Name Version Build Channel

_libgcc_mutex 0.1 conda_forge conda-forge
_openmp_mutex 4.5 2_gnu conda-forge
accelerate 0.22.0
addict 2.4.0
aiohappyeyeballs 2.6.1
aiohttp 3.11.18
aiosignal 1.3.2
albucore 0.0.24
albumentations 2.0.6
aliyun-python-sdk-core 2.16.0
aliyun-python-sdk-kms 2.16.5
annotated-types 0.7.0
anyio 4.9.0
asttokens 3.0.0
async-timeout 5.0.1
asynctest 0.13.0
attrs 25.3.0
blis 0.7.11
bypy 1.8.9
bzip2 1.0.8 h4bc722e_7.conda
ca-certificates 2025.4.26 hbd8a1cb_0.conda
catalogue 2.0.10
certifi 2025.4.26
cffi 1.17.1
charset-normalizer 3.4.2
click 8.1.8
cmake 4.0.0
codecov 2.1.13
colorama 0.4.6
confection 0.1.5
contourpy 1.3.2
coverage 7.8.0
crcmod 1.7
cryptography 44.0.3
cycler 0.12.1
cymem 2.0.11
Cython 3.0.10
datasets 2.17.1
decorator 5.2.1
defusedxml 0.7.1
dill 0.3.8
distro 1.9.0
einops 0.3.2
en-core-web-lg 3.6.0
en-core-web-sm 3.6.0
evaluate 0.3.0
exceptiongroup 1.2.2
executing 2.2.0
filelock 3.14.0
flake8 7.2.0
fonttools 4.57.0
frozenlist 1.6.0
fsspec 2023.10.0
groundingdino 0.1.0
h11 0.16.0
httpcore 1.0.9
httpx 0.27.0
huggingface-hub 0.20.3
idna 3.10
imageio 2.37.0
imgaug 0.4.0
iniconfig 2.1.0
interrogate 1.7.0
ipython 8.18.1
isort 6.0.1
jedi 0.19.2
Jinja2 3.1.6
jmespath 0.10.0
joblib 1.5.0
kiwisolver 1.4.8
kwarray 0.7.1
langcodes 3.5.0
language_data 1.3.0
lanms_neo 1.0.2
lazy_loader 0.4
ld_impl_linux-64 2.43 h712a8e2_4.conda
libexpat 2.7.0 h5888daf_0.conda
libffi 3.4.6 h2dba641_1.conda
libgcc 14.2.0 h767d61c_2.conda
libgcc-ng 14.2.0 h69a702a_2.conda
libgomp 14.2.0 h767d61c_2.conda
liblzma 5.8.1 hb9d3cd8_0.conda
libnsl 2.0.1 hd590300_0.conda
libsqlite 3.49.1 hee588c1_2.conda
libuuid 2.38.1 h0b41bf4_0.conda
libxcrypt 4.4.36 hd590300_1.conda
libzlib 1.3.1 hb9d3cd8_2.conda
lit 18.1.8
lmdb 1.6.2
marisa-trie 1.2.1
Markdown 3.8
markdown-it-py 3.0.0
MarkupSafe 3.0.2
matplotlib 3.7.3
matplotlib-inline 0.1.7
mccabe 0.7.0
mdurl 0.1.2
mmcv 2.0.1
mmdet 3.0.0rc5
mmengine 0.7.1
mmocr 1.0.0
model-index 0.1.11
modelscope 1.25.0
mpmath 1.3.0
multidict 6.4.3
multiprocess 0.70.16
murmurhash 1.0.12
ncurses 6.5 h2d0b736_3.conda
networkx 3.4.2
nltk 3.9.1
numpy 1.24.4
nvidia-cublas-cu11 11.10.3.66
nvidia-cuda-cupti-cu11 11.7.101
nvidia-cuda-nvrtc-cu11 11.7.99
nvidia-cuda-runtime-cu11 11.7.99
nvidia-cudnn-cu11 8.5.0.96
nvidia-cufft-cu11 10.9.0.58
nvidia-curand-cu11 10.2.10.91
nvidia-cusolver-cu11 11.4.0.1
nvidia-cusparse-cu11 11.7.4.91
nvidia-nccl-cu11 2.14.3
nvidia-nvtx-cu11 11.7.91
openai 1.23.1
opencv-python 4.11.0.86
opencv-python-headless 4.11.0.86
opendatalab 0.0.10
openmim 0.3.9
openssl 3.5.0 h7b32b05_1.conda
openxlab 0.1.2
ordered-set 4.1.0
oss2 2.17.0
packaging 24.2
pandas 2.2.3
parameterized 0.9.0
parso 0.8.4
pathlib_abc 0.1.1
pathy 0.11.0
pexpect 4.9.0
pillow 11.2.1
pip 25.1.1
pip 25.1.1 pyh8b19718_0.conda
platformdirs 4.3.8
pluggy 1.5.0
preshed 3.0.9
prompt_toolkit 3.0.51
propcache 0.3.1
psutil 7.0.0
ptyprocess 0.7.0
pure_eval 0.2.3
py 1.11.0
pyarrow 20.0.0
pyarrow-hotfix 0.7
pyclipper 1.3.0.post6
pycocotools 2.0.8
pycodestyle 2.13.0
pycparser 2.22
pycryptodome 3.22.0
pydantic 2.11.4
pydantic_core 2.33.2
pyflakes 3.3.2
Pygments 2.19.1
pyparsing 3.2.3
pytest 8.3.5
pytest-cov 6.1.1
pytest-runner 6.0.1
python 3.10.17 hd6af730_0_cpython.conda
python-dateutil 2.9.0.post0
python-decouple 3.8
pytz 2023.4
PyYAML 6.0.2
RapidFuzz 3.13.0
readline 8.2 h8c095d6_2.conda
regex 2024.11.6
requests 2.28.2
requests-toolbelt 1.0.0
responses 0.18.0
rich 13.4.2
safetensors 0.5.3
scikit-image 0.25.2
scikit-learn 1.6.1
scipy 1.15.2
sentence-transformers 2.2.2
sentencepiece 0.2.0
setuptools 80.1.0 pyhff2d567_0.conda
setuptools 60.2.0
shapely 2.1.0
shortuuid 1.0.13
simsimd 6.2.1
six 1.17.0
smart-open 6.4.0
sniffio 1.3.1
spacy 3.6.1
spacy-legacy 3.0.12
spacy-loggers 1.0.5
srsly 2.5.1
stack-data 0.6.3
stringzilla 3.12.5
supervision 0.25.1
sympy 1.13.1
tabulate 0.9.0
termcolor 3.1.0
terminaltables 3.1.10
thinc 8.1.12
threadpoolctl 3.6.0
tifffile 2025.3.30
timm 0.9.16
tk 8.6.13 noxft_h4845f30_101.conda
tokenizers 0.13.3
tomli 2.2.1
torch 2.0.1
torchaudio 2.0.2
torchvision 0.15.2
tqdm 4.65.2
traitlets 5.14.3
transformers 4.31.0
triton 2.0.0
typer 0.9.4
typing-inspection 0.4.0
typing_extensions 4.13.2
tzdata 2025b h78e105d_0.conda
tzdata 2025.2
ubelt 1.3.7
Unidecode 1.3.6
urllib3 1.26.20
wasabi 1.1.3
wcwidth 0.2.13
wheel 0.45.1 pyhd8ed1ab_1.conda
wheel 0.45.1
xdoctest 1.2.0
xxhash 3.5.0
yapf 0.43.0
yarl 1.20.0

[YUEFANG0211]

hi, do you have any further questions?

您好，抱歉再次打扰您！请问您是在cuda124上安装的pytorch2.0.1吗？我关注到pytoch2.0.1只支持cuda118和117，请问您是如何解决这个问题的呢？感谢拨冗回复！

[sunnychenxiwang]

NVIDIA 驱动支持向下兼容，较新的驱动版本可以支持旧版本的 CUDA 运行时。因此，驱动支持 Torch 2.0.1 所需的 CUDA 11.8。