pip安装pytorch和cuda的经验

原创
2022/02/09 13:21
阅读数 903
AI总结

通过pip安装pytorch

通过pip安装pytorch一般会提示pytorch的名字是torch,安装失败。 解决办法是参照官网工具生成命令行。进入官网(pytorch官网)。

其中run the command即运行的命令,需要注意的是,选择cuda平台需要安装cuda组件。

测试

通过如下命令行测试安装状态:

import torch 
torch.__version__ 
torch.cuda.is_available()

 

展开阅读全文
加载中
点击引领话题📣 发布并加入讨论🔥
0 评论
0 收藏
0
分享
AI总结
返回顶部
顶部