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