本文以pytorch1.5为例,通过下载whl文件的方式,如需安装其他版本,步骤一样
下载pytorch1.5的同时,我们附带下载torchvision0.6.0
①首先查看自己cuda版本,命令ncvv -V即可查看
②作者使用的cuda10.1,查看支持的pytorch版本,cuda10.1支持的torch版本还是比较多的
③然后我选择了pytorch1.5开始正式下载:
pytorch与torchvision下载链接:https://download.pytorch/whl/torch_stable.html
进入下载页面之后,ctrl+f寻找你要下载的pytorch版本,(注意:cu101代表cuda10.1,cp37代表python版本,win_amd64代表是windows系统下载的),找到之后点击下载
④我对应的就是红色箭头指的。同理找torchvision,这个是专门用来处理图像的库,下载比较快
⑤下载完pytorch和torchvision的.whl文件之后,在下载的目录内使用下列命令即可:
pip install torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
pip install torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl
本文以pytorch1.5为例,通过下载whl文件的方式,如需安装其他版本,步骤一样
下载pytorch1.5的同时,我们附带下载torchvision0.6.0
①首先查看自己cuda版本,命令ncvv -V即可查看
②作者使用的cuda10.1,查看支持的pytorch版本,cuda10.1支持的torch版本还是比较多的
③然后我选择了pytorch1.5开始正式下载:
pytorch与torchvision下载链接:https://download.pytorch/whl/torch_stable.html
进入下载页面之后,ctrl+f寻找你要下载的pytorch版本,(注意:cu101代表cuda10.1,cp37代表python版本,win_amd64代表是windows系统下载的),找到之后点击下载
④我对应的就是红色箭头指的。同理找torchvision,这个是专门用来处理图像的库,下载比较快
⑤下载完pytorch和torchvision的.whl文件之后,在下载的目录内使用下列命令即可:
pip install torchvision-0.6.0+cu101-cp37-cp37m-win_amd64.whl
pip install torch-1.5.0+cu101-cp37-cp37m-win_amd64.whl