pytorch历史版本下载

pytorch历史版本下载,第1张

1.下载最新版本的pytorch

https://pytorch.org/官网可以根据 *** 作系统,选择cuda版本, 和选择conda命令还是pip命令:

例如下载最新版本的pytorch命令为:

pip3 install torch torchvision torchaudio --extra-index-url https://download.pytorch.org/whl/cu113
2.pytorch历史版本下载

网址:https://pytorch.org/get-started/previous-versions/

3.下载完成后检测gpu是否可用

代码:

import torch
torch.cuda.is_available()  

输出为ture表示gpu可用

欢迎分享,转载请注明来源:内存溢出

原文地址:https://54852.com/langs/800170.html

(0)
打赏 微信扫一扫微信扫一扫 支付宝扫一扫支付宝扫一扫
上一篇 2022-05-06
下一篇2022-05-06

发表评论

登录后才能评论

评论列表(0条)

    保存