win10 安装torch2.0 + GPU

win10 安装 torch==2.0.0+cu11 失败

pip install torch==2.0.0+cu118 torchvision==0.15.1+cu118 -f https://download.pytorch.org/whl/torch_stable.html -i https://pypi.douban.com/simple/ --trusted-host=pypi.douban.com/simple

通过上述命令安装一直报错:

ERROR: Wheel 'torch' located at C:\...

可以采用下载 whl 到本地安装:到 地址 下载对应版本 whl 到本地,如:torch-2.0.0+cu118-cp310-cp310-win_amd64.whl。然后通过 pip install torch-2.0.0+cu118-cp310-cp310-win_amd64.whl 安装,注意这里 whl 要用其所在路径。

备份一个 LLM langchain 的 requirements.txt。

aiohttp==3.8.5
aiosignal==1.3.1
async-timeout==4.0.2
attrs==23.1.0
certifi==2023.7.22
charset-normalizer==3.2.0
click==8.1.6
colorama==0.4.6
dataclasses-json==0.5.13
faiss-cpu==1.7.4
filelock==3.12.2
frozenlist==1.4.0
fsspec==2023.6.0
greenlet==2.0.2
huggingface-hub==0.16.4
idna==3.4
Jinja2==3.1.2
joblib==1.3.1
langchain==0.0.240
langsmith==0.0.14
MarkupSafe==2.1.3
marshmallow==3.20.1
mpmath==1.3.0
multidict==6.0.4
mypy-extensions==1.0.0
networkx==3.1
nltk==3.8.1
numexpr==2.8.4
numpy==1.25.1
openapi-schema-pydantic==1.2.4
packaging==23.1
Pillow==10.0.0
pydantic==1.10.11
PySocks
PyYAML==6.0.1
regex==2023.6.3
requests==2.31.0
safetensors==0.3.1
scikit-learn==1.3.0
scipy==1.11.1
sentence-transformers==2.2.2
sentencepiece==0.1.99
SQLAlchemy==2.0.19
sympy==1.12
tenacity==8.2.2
threadpoolctl==3.2.0
tokenizers==0.13.3
torch====2.0.0+cu118
torchvision==0.15.1+cu118
tqdm==4.65.0
transformers==4.31.0
typing-inspect==0.9.0
typing_extensions==4.7.1
urllib3==2.0.4
win-inet-pton
yarl==1.9.2

正文完
 
admin
版权声明:本站原创文章,由 admin 2023-11-26发表,共计1369字。
转载说明:除特殊说明外本站文章皆由CC-4.0协议发布,转载请联系tensortimes@gmail.com。
评论(没有评论)
验证码