CUDA12.1安装对应的Pytorch版本

1. 创建新环境安装Python版本为3.9.7
2. 使用命令安装Pytorchpip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
3. 出现超时无法下载在pip的配置文件中设置超时时间，下边为默认配置文件路径

C:\Users\***\AppData\Roaming\pip

其中***为当前计算机登录的用户名

下边为配置文件内容，使用了清华镜像源

[global]
timeout = 600000
index-url = https://pypi.tuna.tsinghua.edu.cn/simple

进行transformers安装时建议安装4.26.1版本

pip install transformers==4.26.1