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