Skip to content

Change the installed torch version after installing Compressor AI #342

@zb12138

Description

@zb12138

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

No one assigned

    Labels

    No labels
    No labels

    Type

    No type

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions