为PyTorch 1.10.0(CUDA11.0)适配安装torchaudio求助
我的Ubuntu 18.04机器上有一个虚拟环境,其中包含pytorch=1.10.0=cuda110py38hf84197b_0,通过nvidia-smi确认CUDA版本为11.0,想要安装torchaudio。
尝试mamba install torchaudio=0.10.0 -c pytorch时,PyTorch的CUDA版本会从11.0升级到11.2;安装torchaudio=0.9.1则会把PyTorch版本从1.10.0降级到1.9.1。
这台机器运行多年旧代码,依赖这些旧包,希望尽可能少修改包或CUDA驱动来安装torchaudio。是否有办法为PyTorch 1.10.0(CUDA11.0版本)安装torchaudio?查看PyTorch官方历史版本页面似乎没有相关选项,但仍想寻求解决办法。
以下是PyTorch相关基础环境配置:
name: torchbase channels: - pytorch - conda-forge - defaults dependencies: - _libgcc_mutex=0.1=conda_forge - _openmp_mutex=4.5=1_llvm - bzip2=1.0.8=h7f98852_4 - ca-certificates=2023.11.17=hbcca054_0 - cffi=1.15.0=py38h3931269_0 - cudatoolkit=11.0.3=h15472ef_9 - cudnn=8.2.1.32=h86fa8c9_0 - future=0.18.2=py38h578d9bd_4 - icu=68.2=h9c3ff4c_0 - ld_impl_linux-64=2.36.1=hea4e1c9_2 - libblas=3.9.0=12_linux64_mkl - libcblas=3.9.0=12_linux64_mkl - libffi=3.4.2=h7f98852_5 - libgcc-ng=11.2.0=h1d223b6_11 - libiconv=1.16=h516909a_0 - liblapack=3.9.0=12_linux64_mkl - libnsl=2.0.0=h7f98852_0 - libprotobuf=3.18.1=h780b84a_0 - libstdcxx-ng=11.2.0=he4da1e4_11 - libuuid=2.32.1=h7f98852_1000 - libxml2=2.9.12=h72842e0_0 - libzlib=1.2.11=h36c2ea0_1013 - llvm-openmp=12.0.1=h4bd325d_1 - magma=2.5.4=h4a2bb80_2 - mkl=2021.4.0=h8d4b97c_729 - nccl=2.11.4.1=h96e36e3_0 - ncurses=6.2=h58526e2_4 - ninja=1.10.2=h4bd325d_1 - numpy=1.22.3=py38h99721a1_2 - openssl=1.1.1o=h166bdaf_0 - pip=23.3.2=pyhd8ed1ab_0 - pycparser=2.21=pyhd8ed1ab_0 - python=3.8.12=hb7a2778_2_cpython - python_abi=3.8=2_cp38 - pytorch=1.10.0=cuda110py38hf84197b_0 - pytorch-gpu=1.10.0=cuda110py38h5b0ac8e_0 - readline=8.1=h46c0cb4_0 - setuptools=60.1.1=py38h578d9bd_0 - sleef=3.5.1=h9b69904_2 - tbb=2021.5.0=h4bd325d_0 - tk=8.6.11=h27826a3_1 - typing_extensions=4.0.1=pyha770c72_0 - wheel=0.37.1=pyhd8ed1ab_0 - xz=5.2.5=h516909a_1 - zlib=1.2.11=h36c2ea0_1013 - zstd=1.5.1=ha95c52a_0
指定严格版本约束,强制保留PyTorch和CUDA版本
使用mamba安装时,通过==锁定PyTorch和cudatoolkit的完整版本,避免被升级或降级:mamba install torchaudio=0.10.0 pytorch==1.10.0=cuda110py38hf84197b_0 cudatoolkit==11.0.3 -c pytorch这个命令会强制mamba满足指定的PyTorch和CUDA版本,仅寻找兼容的torchaudio包。
从wheel包手动安装
如果conda/mamba找不到兼容包,直接用pip安装对应版本的torchaudio wheel文件,明确指定CUDA 11.0后缀:pip install torchaudio==0.10.0+cu110 -f https://download.pytorch.org/whl/torch_stable.html该方式不会触发PyTorch版本变更,直接匹配现有环境的CUDA和Python版本。
限制conda搜索通道
当前环境通道顺序包含conda-forge,可能引入版本冲突,尝试仅从pytorch通道搜索:mamba install torchaudio=0.10.0 -c pytorch --override-channels
内容的提问来源于stack exchange,提问作者Brian Provost

