0.4.0pytorch运行过程中对0-dim和volatile提示UserWarning的解决方法

15 篇文章 25 订阅

将pytorch更新到0.4.0最新版后对0.3.1版本代码会有如下警告,它在提醒用户下个版本这将成为一个错误

1.

UserWarning: invalid index of a 0-dim tensor. This will be an error in PyTorch 0.5. Use tensor.item() to convert a 0-dim tensor to a Python number
  train_loss += loss.data[0]

修改如下:

#原语句:
train_loss+=loss.data[0]
#修改后:
train_loss+=loss.item()
#bingo

2.

UserWarning: volatile was removed and now has no effect. Use `with torch.no_grad():` instead.
  label = Variable(label.cuda(), volatile=True)

修改如下:

#原语句
label = Variable(label.cuda(), volatile=True)
#修改后语句
  label = Variable(label.cuda())

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
### 回答1: pytorch 0.4.0版本与cudatoolkit 9兼容。cudatoolkit是一个用于深度学习和科学计算的加速库,它提供了与NVIDIA GPU一起使用的工具和库。pytorch 0.4.0版本要求使用cudatoolkit 9或更高版本来发挥最佳性能。 另外,pytorch 0.4.0版本也要求使用mkl(Math Kernel Library)的2018或更高版本。mkl是针对数值计算优化的数学函数库,它提供高性能的数学函数和线性代数运算,可以加速深度学习算法的计算过程。 通过将pytorch 0.4.0与cudatoolkit 9和mkl 2018或更高版本进行匹配,可以利用GPU加速深度学习任务,并且使用高性能的数学函数库来提高计算效率。这些更新和要求可以提供更好的训练和推断性能,使得深度学习的计算更加高效和快速。因此,在使用pytorch 0.4.0版本时,务必安装和配置正确的cudatoolkit和mkl版本以获得最佳的性能和功能。 ### 回答2: PyTorch 0.4.0版本要求的依赖为Cuda Toolkit 9及以上版本和MKL 2018及以上版本。 Cuda Toolkit是NVIDIA提供的用于支持GPU加速的工具集。PyTorch使用了Cuda来充分发挥GPU的计算能力,提高模型训练和推理的速度。因此,安装PyTorch 0.4.0版本时需要安装Cuda Toolkit 9及以上版本。 MKL(Math Kernel Library)是Intel提供的数学库,具有高性能的线性代数、FFT和随机数生成等功能。在PyTorch,MKL用于加速模型的数值计算。版本2018及以上的MKL提供了更好的性能和稳定性。因此,安装PyTorch 0.4.0版本时需要安装MKL 2018及以上版本。 通过安装Cuda Toolkit和MKL,可以使PyTorch在GPU上充分利用并行计算和高效的数学库,提高模型的训练和推理速度。 ### 回答3: pytorch 0.4.0版本在安装时有两种依赖方式可以选择:cudatoolkit 9.* 和mkl >=2018。 首先,cudatoolkit是用于支持GPU加速的工具包,版本9.*表示使用任何9.x版本的cudatoolkit都可以。如果你希望在安装pytorch时开启GPU加速,需要先安装对应版本的cudatoolkit。 其次,mkl是Intel提供的数学库,可以提升计算效率。在安装pytorch时,通过指定mkl >=2018依赖,可以确保安装的pytorch版本能够充分利用mkl库的功能。 综上所述,当你安装pytorch 0.4.0*版本时,可以根据你的需求选择安装cudatoolkit 9.*或者mkl >=2018。如果你需要GPU加速,选择安装对应版本的cudatoolkit;如果你需要提升计算效率,安装最新版本的mkl库。

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值