Pytorch离线安装教程
我们总共需要4个文件:
下载地址
链接:https://pan.baidu.com/s/1cUMFYLF3CVfHmXc1hUZrbA
提取码:nqoa
步骤一:安装两个包
打开anaconda prompt
列出环境
conda env list
删除环境
conda env remove -n (你要删的环境)
新建环境
conda create -n pytorch16 python=3.8
激活环境
conda activate pytorch16
找到安装包存放的路径
通过cd命令去到存放包的目录下
在安装离线包的时候同时会下载相关的包,我们添加镜像源加快下载速度:
1 | pip config set global.index-url https://mirrors.aliyun.com/pypi/simple |
然后执行下面命令:
1 | pip install torch-1.6.0+cu101-cp38-cp38-win_amd64.whl |
然后继续安装第二个包:
1 | pip install torchvision-0.7.0+cu101-cp38-cp38-win_amd64.whl |
步骤二:安装显卡驱动
下载地址
https://developer.nvidia.com/cuda-10.1-download-archive-base
如果之前安装过cuda要先卸载,卸载方法如下:
然后删除文件夹:
下载好之后安装(全部默认下一步)
下载cudnn:
然后解压cudnn,将里面的内容复制到cuda文件夹
步骤三:验证安装成功
1 | import torch |
输出
1 | 1.6.0+cu101 |
成功