安装pytorch，使用transformers

静听山水

已于 2024-07-29 16:52:37 修改

阅读量409

点赞数

CC 4.0 BY-SA版权

分类专栏：深度学习文章标签：机器学习

于 2023-11-07 13:44:53 首次发布

本文链接：https://blog.youkuaiyun.com/qq_41081716/article/details/134265375

深度学习专栏收录该内容

11 篇文章

订阅专栏

文章介绍了在Mac环境中使用Conda创建和激活npytorch_env和torch_contract环境，安装PyTorch及其依赖（如torchvision,torchaudio,numpy），以及额外安装Rust和transformers库的过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Mac：

conda create -n pytorch_env python=3.9.16
conda activate pytorch_env
conda install pytorch torchvision torchaudio -c pytorch
 
pip install rust
pip install tokenizers
pip install transformers




import torch 
print(torch.cuda.is_available())

Linux：

conda create --name torch_contract python=3.9.16
conda activate torch_contract
pip3 install numpy --pre torch --force-reinstall --index-url https://download.pytorch.org/whl/nightly/cu117

pip3 install numpy --pre torch --force-reinstall --index-url https://pypi.tuna.tsinghua.edu.cn/simple

pip install transformers