Pytorch1.12.1+cu113安装记录

因为torch1.7.0对于SiLU算子导出不支持,需要1.7.1才支持.于是索性准备更新一下自己的算法版本库,查询到CUDA11.3支持的最高Pytorch版本为1.12.1,于是统一做一下更新.这里采用离线下载的方式,因为在线下载实在是太蛋疼了

1. Pytorch离线包下载

这是官方提供的版本,我这里不直接安装,而是先去下载这些包

# CUDA 11.3
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113

下载地址:

Pytorch包下载官方地址

torch下载链接:

http://download.pytorch.org/whl/cu113/torch-1.12.1%2Bcu113-cp39-cp39-linux_x86_64.whl

torchvision下载链接:

http://download.pytorch.org/whl/cu113/torchvision-0.13.1%2Bcu113-cp39-cp39-win_amd64.whl

torchaudio下载链接:

http://download.pytorch.org/whl/cu113/torchaudio-0.12.1%2Bcu113-cp39-cp39-win_amd64.whl

2.CUDA及cudnn包下载

cuda11.3下载地址

cudnn下载地址

3.测试

4.安装好的环境迁移到别的电脑

我想试试把env/里面的整个环境包打包发送到别的电脑上,看能否直接调用,理论上是可行的.因为每个环境内都有单独的python.exe

 

实测可以直接调用并找到环境中的包,但是使用anaconda切换环境会有未知的错误.可以使用 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值