因为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切换环境会有未知的错误.可以使用