pytorch(GPU) 安装简记—Windows版本

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

image-20200622170823193.png

③:查看pytorch的whl文件

pytorch官网:https://pytorch.org/

image-20200622171340125.png

如果你的cuda版本是上图这些,或者是CPU版本,直接参见上图选择下载命令。

否则:

若你的cuda是其他版本,请点击上图绿框,选择满足你cuda的pytorch版本。

④:下载pytorch的whl文件

查看的历史pytorch版本,如下图所示:

image-20200622173337471.png

我的cuda版本是9.0,所以复制上图红色框的连接地址并打开(注意根据自己的cuda版本去选地址),如下图示:(选择点击下载)

image-20200622174008120.png

注意:你需要注意你的python版本,和Windows版本是64位。

⑤:安装pytorch

首先:cd 到下载文件的目录

然后:命令:pip install your_pytorch_file_name

⑥:安装torchvision

下载链接同样是④那个页面中的,图示如下:

image-20200622175232701.png

首先:cd 到下载文件的目录

然后:命令:pip install your_pytorchvision_file_name

3. 安装成功

检查是否安装成功,如果出现下图说明安装成功,一切ok!

image-20200622190347113.png

4. 注意事项

1,python 环境不可 3.6.0,

2,选择下载自己cuda对应的pytorch版本

©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。