记录Pytorch安装的一些流程
Pytorch官网:https://pytorch.org/get-started/locally/
本文记录安装Pytorch的流程,不包括安装cuda驱动
Pytorch官网下载速度太慢,尝试多次不成功
因此推荐离线安装,步骤如下:
-
根据pytorch官网的命令选择安装版本,如下图,记录需要安装的版本
-
下载对应版本pytorch的whl文件:https://download.pytorch.org/whl/(该网站在国内有时打不开,需要多次尝试)
- 下载torch, torchvision就足够了
- 点击torch,用 Crtl+F 搜索对应版本
- 文件命名规则,如 torch-1.9.0+cu102-cp38-cp38-win_amd64.whl,表示1.9.0版本的pytorch,支持cuda10.2和python3.8,window64位系统
- 下载的pytorch版本要和电脑中python的版本相对应,否则无法安装
- torchvision下载同理
- 下载完成后,运行Anaconda Powershell Prompt (Anaconda)
依次输入命令pip install <whl文件存放路径>
稍等片刻便安装成功