栏目分类:
子分类:
返回
名师互学网用户登录
快速导航关闭
当前搜索
当前分类
子分类
实用工具
热门搜索
名师互学网 > IT > 软件开发 > 后端开发 > Python

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

Python 更新时间: 发布时间: IT归档 最新发布 模块sitemap 名妆网 法律咨询 聚返吧 英语巴士网 伯小乐 网商动力

深度学习入门篇——一、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 产品的新驱动。https://www.nvidia.cn/Download/index.aspx?lang=cn#

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

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

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

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

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

 安装过程:

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

②在开始菜单栏打开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()

转载请注明:文章转载自 www.mshxw.com
本文地址:https://www.mshxw.com/it/969449.html
我们一直用心在做
关于我们 文章归档 网站地图 联系我们

版权所有 (c)2021-2022 MSHXW.COM

ICP备案号:晋ICP备2021003244-6号