Skip to content

Change the installed torch version after installing Compressor AI #342

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

Activity

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

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