1. 安装前准备
①:安装 Anaconda (强烈推荐!!!)
地址:anaconda:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
minicondadi:https://mirrors.tuna.tsinghua.edu.cn/anaconda/miniconda/
②:安装符合你显卡cuda和cudnn
若以前使用配置过GPU,可以忽略此步骤。
2. 开始安装
①:使用conda创建虚拟环境
命令:conda create --name torch python==3.6.X (注意:这里一定要高于3.6.0版本,否则后期安装pytorch报错!!!)
安装成功后激活环境:activate torch (以后都在该环境里操作)
②:查看你的cuda版本(图示如下)
命令:nvcc -V
③:查看pytorch的whl文件
pytorch官网:https://pytorch.org/
如果你的cuda版本是上图这些,或者是CPU版本,直接参见上图选择下载命令。
否则:
若你的cuda是其他版本,请点击上图绿框,选择满足你cuda的pytorch版本。
④:下载pytorch的whl文件
查看的历史pytorch版本,如下图所示:
我的cuda版本是9.0,所以复制上图红色框的连接地址并打开(注意根据自己的cuda版本去选地址),如下图示:(选择点击下载)
注意:你需要注意你的python版本,和Windows版本是64位。
⑤:安装pytorch
首先:cd 到下载文件的目录
然后:命令:pip install your_pytorch_file_name
⑥:安装torchvision
下载链接同样是④那个页面中的,图示如下:
首先:cd 到下载文件的目录
然后:命令:pip install your_pytorchvision_file_name
3. 安装成功
检查是否安装成功,如果出现下图说明安装成功,一切ok!
4. 注意事项
1,python 环境不可 3.6.0,
2,选择下载自己cuda对应的pytorch版本