pytorch1.5+torchvision0.6.0安装教程(win10)

本文以pytorch1.5为例,通过下载whl文件的方式,如需安装其他版本,步骤一样

下载之前查看自己cuda版本,命令ncvv -V即可查看

pytorch1.5+torchvision0.6.0安装教程(win10)

作者使用的cuda10.1,下表为cuda与torch对应表,cuda10.1支持的torch版本还是比较多的

pytorch1.5+torchvision0.6.0安装教程(win10)

然后开始正式下载:

pytorch与torchvision下载链接:https://download.pytorch.org/whl/torch_stable.html

注意:cu101代表cuda10.1,cp37代表python版本,win_amd64代表是windows系统下载的

pytorch1.5+torchvision0.6.0安装教程(win10)

我对应的就是红色箭头指的,同理下载torchvision,这个是专门用来处理图像的库,下载比较快

pytorch1.5+torchvision0.6.0安装教程(win10)

下载完之后,在下载的目录内使用下列命令即可:

pip install torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
pip install torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl
上一篇:2021-11-02 关于lxml安装报错问题


下一篇:Python 大白从零开始 OpenCV 学习课-1.安装与环境配置