-
Notifications
You must be signed in to change notification settings - Fork 265
Open
Description
Bug
I plan to use pytorch 1.31+python 3.10 + cuda 11.7 in conda env. However, when I used pip to install compressai, it automatically downloaded and installed cuda12 and torch 2.8.0, and installing from the source code got the same result. This really confuses me. Did compressai stop supporting other versions of pytorch?
To Reproduce
conda create -n env python=3.10
pip install torch==1.13.1+cu117 torchvision==0.14.1+cu117 torchaudio==0.13.1 --extra-index-url https://download.pytorch.org/whl/cu117
pip install compressai
pip show torch
Expected behavior
keep torch=1.13.1
Environment
Please copy and paste the output from python3 -m torch.utils.collect_env
Collecting environment information...
PyTorch version: 1.13.1+cu117
Is debug build: False
CUDA used to build PyTorch: 11.7
ROCM used to build PyTorch: N/A
OS: Ubuntu 22.04.3 LTS (x86_64)
GCC version: (Ubuntu 7.5.0-6ubuntu2) 7.5.0
Clang version: Could not collect
CMake version: version 3.22.1
Libc version: glibc-2.35
Python version: 3.10.18 (main, Jun 5 2025, 13:14:17) [GCC 11.2.0] (64-bit runtime)
Python platform: Linux-5.15.123.1-microsoft-standard-WSL2-x86_64-with-glibc2.35
Is CUDA available: True
CUDA runtime version: 11.7.64
CUDA_MODULE_LOADING set to: LAZY
GPU models and configuration:
GPU 0: NVIDIA GeForce RTX 4090
GPU 1: NVIDIA GeForce RTX 3090
Nvidia driver version: 560.94
cuDNN version: Could not collect
HIP runtime version: N/A
MIOpen runtime version: N/A
Is XNNPACK available: True
Versions of relevant libraries:
[pip3] numpy==1.26.4
[pip3] pytorch-msssim==1.0.0
[pip3] torch==1.13.1+cu117
[pip3] torch-geometric==2.6.1
[pip3] torch_scatter==2.1.2
[pip3] torchaudio==0.13.1+cu117
[pip3] torchvision==0.14.1+cu117
[conda] numpy 1.26.4 pypi_0 pypi
[conda] pytorch-msssim 1.0.0 pypi_0 pypi
[conda] torch 1.13.1+cu117 pypi_0 pypi
[conda] torch-geometric 2.6.1 pypi_0 pypi
[conda] torch-scatter 2.1.2 pypi_0 pypi
[conda] torchaudio 0.13.1+cu117 pypi_0 pypi
[conda] torchvision 0.14.1+cu117 pypi_0 pypi
Metadata
Metadata
Assignees
Labels
No labels