深度学习入门篇——一、Pytorch环境配置(GPU:GeForce 940MX)

安装pytorch的几点注意事项:

①确定你的电脑显卡是否支持cuda。可取网站查询:CUDA Zone - Library of Resources | NVIDIA Developerhttps://www.geforce.com/hardware/technology/cuda/supported-gpus

    确保你的显卡驱动更新到最新,以保证能兼容最新版本的cuda。     官方驱动 | NVIDIA下载适用于 GeForce、TITAN、NVIDIA RTX、数据中心、GRID 等 NVIDIA 产品的新驱动。icon-default.png?t=M4ADhttps://www.nvidia.cn/Download/index.aspx?lang=cn#

②确认显卡支持的cuda版本,很重要!!不然之后下载会因为版本问题导致一系列问题。

查询方法:打开电脑NVIDIA控制面板 — 帮助 — 系统信息 — 显示 — 组件。

深度学习入门篇——一、Pytorch环境配置(GPU:GeForce 940MX)_第1张图片深度学习入门篇——一、Pytorch环境配置(GPU:GeForce 940MX)_第2张图片

查询到你显卡所支持的cuda版本,再去pytorch网站找对应的cuda版本下载命令。

Previous PyTorch Versions | PyTorchhttps://pytorch.org/get-started/previous-versions/

深度学习入门篇——一、Pytorch环境配置(GPU:GeForce 940MX)_第3张图片

 ③搭建环境时,确保anaconda支持的python版本与你所搭建的pytorch中的python版本一致。

深度学习入门篇——一、Pytorch环境配置(GPU:GeForce 940MX)_第4张图片

 安装过程:

 ①上anaconda官网下载对应版本的安装包进行安装。此时,需要记住你自定义的安装路径。

深度学习入门篇——一、Pytorch环境配置(GPU:GeForce 940MX)_第5张图片

②在开始菜单栏打开anaconda prompt命令行窗口,输入

conda create -n pytorch python=3.7

 这里的python版本建议与anaconda对应的python版本一致。

③创建pytorch环境后,输入以下代码来激活所创建的环境。

conda activate pytorch

 ④将前述的显卡对应的cuda版本下载代码复制进去,回车开始安装所需的各种package。

conda install pytorch==1.1.0 torchvision==0.3.0 cudatoolkit=9.0 -c pytorch

等待安装完成。

⑤验证安装成功。依次输入以下代码,查看是否报错,如果不报错,即安装成功。

python


import torch


torch.cuda.is_available()

你可能感兴趣的:(pytorch,深度学习,人工智能)