安装torch(非pytorch)

本文档详细介绍了在Ubuntu 16.04系统上安装cuda9.0、gcc5.0和cudnn7后,如何安装非PyTorch的Torch。在安装过程中,由于cuda9.0对应cudnn6以上版本,而Torch默认支持cudnn5.x,导致可能的版本不兼容问题。解决方法包括修改Torch的cudnn版本。安装步骤包括git克隆Torch仓库,安装依赖,设置编译标志并执行安装脚本。安装完成后需设置环境变量。此外,还提到了安装过程中可能出现的Ubuntu源问题和bash install.sh时的错误及其解决策略。
摘要由CSDN通过智能技术生成

安装torch(非pytorch)

系统环境ubuntu16.04+cuda9.0+gcc5.0+cudnn7

*cuda9只对应cudnn6以上版本,但是torch默认支持cudnn5.x版本,所以会出现版本不兼容问题,后面会写如何解决,如果不想出现该问题,可以直接安装cuda8.0+cudnn5(没试过,应该可以)

git clone https://github.com/torch/distro.git ~/torch –recursive

cd ~/torch

bash install-deps

export TORCH_NVCC_FLAGS="-D__CUDA_NO_HALF_OPERATORS__"

bash install.sh

安装编译需要一会时间。

安装好后,会提示设置环境变量,输入yes就行。

然后执行source ~/.bashrc激活相应的环境变量。

这是在终端输入th就可以呼出torch的控制界面:

注意事项

1、torch默认支持cudnn5,使用cudnn7会报如下错误

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值