装特定的torch和cuda，cu118

强化学习曾小健

已于 2024-08-14 23:11:43 修改

阅读量2.4k

点赞数 5

分类专栏： # Pytorch源码解析文章标签：深度学习人工智能

于 2024-07-15 11:00:36 首次发布

本文链接：https://blog.csdn.net/sinat_37574187/article/details/140433042

版权

pip install torch==2.2.0+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

cuda12.4

pip install torch==2.4.0+cu124 --extra-index-url https://download.pytorch.org/whl/cu124

LMdeploy所需要的最低torch

pip install torch==2.2.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

vllm 0.4.0-cu118

torch==2.1.2

pip install torch==2.1.2+cu118 --extra-index-url https://download.pytorch.org/whl/cu118

Requirement already satisfied: filelock in ./mambaforge/envs/Qwen/lib/python3.12/site-packages (from torch==2.2.0+cu118) (3.13.4)
Requirement already satisfied: typing-extensions>=4.8.0 in ./mambaforge/envs/Qwen/lib/python3.12/site-packages (from torch==2.2.0+cu118) (4.11.0)
Requirement already satisfied: sympy in ./mambaforge/envs/Qwen/lib/python3.12/site-packages (from