安装xFormers时遇到的问题,以及正确的安装方式

在使用pip install xformers安装xformers时,发现总是会把我环境中的pytorch重新安装,并且会安装CUDA12版本的pytorch, 而我环境是CUDA 11.8,这就导致我原本的开发环境不可用了。后来发现xformers与pytorch版本一一对应的,在pip install xformers时,如果发现pytorch版本不一致,就会自动卸载重装pytorch, 而默认装的是CUDA12版本的。
因此可以去https://github.com/facebookresearch/xformers查看最新xformers对应的pytorch版本,预先安装,并根据自己环境的CUDA版本选用以下方式进行安装:

# cuda 11.8 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu118
# cuda 12.1 version
pip3 install -U xformers --index-url https://download.pytorch.org/whl/cu121

相关推荐

  1. 安装xFormers遇到问题以及正确安装方式

    2024-06-15 06:50:05       25 阅读

最近更新

  1. docker php8.1+nginx base 镜像 dockerfile 配置

    2024-06-15 06:50:05       98 阅读
  2. Could not load dynamic library ‘cudart64_100.dll‘

    2024-06-15 06:50:05       106 阅读
  3. 在Django里面运行非项目文件

    2024-06-15 06:50:05       87 阅读
  4. Python语言-面向对象

    2024-06-15 06:50:05       96 阅读

热门阅读

  1. 8个常用的辅助函数!!

    2024-06-15 06:50:05       25 阅读
  2. Cargo

    2024-06-15 06:50:05       33 阅读
  3. Docker从容器打包镜像到本地保存与加载

    2024-06-15 06:50:05       27 阅读
  4. TensorFlow编程环境:构建深度学习的乐园

    2024-06-15 06:50:05       26 阅读
  5. 计算机网络(6) ICMP协议

    2024-06-15 06:50:05       24 阅读