cuda安装精简

使用torch的一定注意cuda版本必须要跟torch版本对应,网上好多网友写的博客都没提到,目前cuda官网的版本是11.0但是我当时在torch官网下载的是支持cuda10.2版本的torch。
在这里插入图片描述
cuda 10.2 链接是cuda10.2版本的下载链接,下载之后按照提示步骤一步步安装即可非常简单。

两种方法测试是否安装成功:
1.进入cmd,输入nvcc -V,会出现cuda的版本号等相关信息。
2.如果正在用torch的话,直接输torch.cuda.is_available(),如果返回True则表示cuda安装成功,我们可以用GPU训练神经网络。

  • 3
    点赞
  • 3
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值