安装特定版本的torch-linux

在尝试运行特定项目时遇到'torch.device'属性缺失的错误,由于服务器CUDA版本为9.0,而最新torch版本最低支持CUDA 9.2。通过回退到torch 0.4版本以上,避免过高版本导致的兼容性问题。从pytorch官网找到适用于CUDA 9.0的torch 1.1.0版本进行下载并升级,最终解决了报错,成功运行程序。

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

尝试运行 https://github.com/jadore801120/attention-is-all-you-need-pytorch 时出错,针对torch版本进行了修改。

torch官网上最新版本的torch支持的cuda最低版本为92,服务器上cuda版本为9.0,如果安装最新版的torch运行时会报错

The NVIDIA driver on your system is too old (found version 9010).

因此只能安装以前的版本的torch

一开始安装了0.3version的torch,torch版本的查看方法是:

import torch

print(torch.__version__)

在运行的时候报错

module 'torch' has no attribute 'device'

原因是torch版本太老了,要升级torch,但是还不能升级到最高版,因为cuda跟不上

为解决这个问题,torch至少要升级到0.4version以上,建议1.0以上吧,要不然还有其他的问题。

https://pytorch.org/get-started/previous-version

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值