pytorch调用geforce rtx 3060

• 825 次浏览 • 0个评论 • 2021年10月24日

想在pytorch中调用GeForce RTX 3060训练模型。 按照pytorch官网提供的代码:

conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch

在 conda 安装了 torch 1.10.0 和 cudatoolkit 11.3.1。

但是运行

torch.cuda.is_available()

之后仍然显示 False。

请问该如何解决啊 :)

提交评论

请登录后评论

用户评论

    当前暂无评价,快来发表您的观点吧...

更多相关好文

    当前暂无更多相关好文推荐...

    公告提示

    Pytorch是Facebook的AI研究团队发布了一个Python工具包，是Python优先的深度学习框架。作为numpy的替代品；使用强大的GPU能力，提供最大的灵活性和速度,实现了机器学习框架Torch在Python语言环境的执行,基于python且具备强大GPU加速的张量和动态神经网络。