torch+cuda gpu并行计算

buguanlan

于 2022-10-20 14:15:02 发布

阅读量933

点赞数 1

CC 4.0 BY-SA版权

文章标签： pytorch 深度学习人工智能

版权声明：本文为博主原创文章，遵循 CC 4.0 BY-SA 版权协议，转载请附上原文出处链接和本声明。

本文链接：https://blog.youkuaiyun.com/buguanlan/article/details/127425798

本文详细介绍了如何根据GPU算力和驱动版本选择合适的CUDA版本，以GT730显卡为例，选择了CUDA 8.0。接着，讲述了安装CUDA并设置环境变量的步骤，最后强调了根据CUDA版本选择对应PyTorch版本进行安装的重要性。提供了一个具体的安装教程链接以供参考。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

torch+ CUDA gpu 并行计算环境配置
需要注意 GPU算力值 cuda版本显卡驱动版本 torch版本之间的匹配

torch版本可以根据cuda版本来
先根据驱动版本和显卡算力（显卡算力自己查）选择合适cuda版本

我的显卡GT730 算力3.5 驱动最高支持版本11.4 选择cuda版本8.0

然后安装下载安装cuda 并且配置环境cuda环境变量

最后根据cuda版本选择pytorch版本，并安装

以上是安装torch+cuda的步骤具体实例安装请参考：https://blog.youkuaiyun.com/Mind_programmonkey/article/details/99688839

评论 1

被折叠的条评论为什么被折叠?

到【灌水乐园】发言

查看更多评论

添加红包

成就一亿技术人!

hope_wisdom

发出的红包

实付元

使用余额支付

点击重新获取

扫码支付

钱包余额 0

抵扣说明：

1.余额是钱包充值的虚拟货币，按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载，可以购买VIP、付费专栏及课程。