ubuntu 安装 pytorch (ubuntu16.04+pytorch0.4.1+python2.7+cuda8.0)

本文介绍了如何在Ubuntu 16.04上使用pip安装PyTorch 0.4.1和CUDA 8.0。由于直接使用CUDA安装不兼容,作者推荐通过pip简单快捷地完成安装。详细步骤包括在PyTorch官网选择合适的配置,复制并执行安装命令,最后验证安装是否成功。

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

这里使用pip安装pytorch,我试过cuda安装,和电脑的配置没兼容,没有安装成功,后来发现使用pip安装很简单方便,就是用pip安装
首先进入pytorch官网,往下拉会看到不同的安装选项
根据自己电脑的配置选择不同的选项,例如我选择的stable版本、Linux系统、pip安装、python版本为2.7、cuda为8.0,如下图所示:
在这里插入图片描述
选择完之后下面会出现安装所需的命令,如下图
在这里插入图片描述
先后运行前两个命令就可以,网速慢的话只能慢慢等了,安装成功第一条命令后是这样的
在这里插入图片描述
安装成功第二条命令后是这样的
在这里插入图片描述
下面测试是否安装成功
命令行中输入

python
import torch
torch.__version__

没有问题的话会输出版本为0.4.1,如下图所示
在这里插入图片描述

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值