CUDA12.1安装对应的Pytorch版本
下边为配置文件内容,使用了清华镜像源。
文章共222字 · 阅读需要大约1分钟
一键AI生成摘要,助你高效阅读
问答
·
- 创建新环境安装Python版本为3.9.7
- 使用命令安装Pytorch
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
- 出现超时无法下载在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
更多推荐
已为社区贡献1条内容
所有评论(0)