主页 > 软件开发  > 

win11编译pytorchvisioncuda128版本流程

win11编译pytorchvisioncuda128版本流程
1. 前置条件

本篇续接自 win11编译pytorch cuda128版本流程,阅读前请先参考上一篇配置环境。

访问 kkgithub /pytorch/vision/archive/refs/tags/v0.21.0.tar.gz下载源码,下载后解压。

2.编译

打开Miniforge Prompt,依次执行如下代码:

conda activate torch_build_env conda install libpng libjpeg-turbo -c pytorch cd c:\vision-0.21.0 ##执行前先确认删除build目录 python setup.py bdist_wheel

 

编译完成后,wheel包在dist目录下,使用pip安装

pip install dist\torchvision-0.21.0-cp310-cp310-win_amd64.whl
标签:

win11编译pytorchvisioncuda128版本流程由讯客互联软件开发栏目发布,感谢您对讯客互联的认可,以及对我们原创作品以及文章的青睐,非常欢迎各位朋友分享到个人网站或者朋友圈,但转载请说明文章出处“win11编译pytorchvisioncuda128版本流程