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

Pytorch环境搭建

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

Pytorch环境搭建

1.创建环境

首先打开Anaconda 命令提示行,即打开Anaconda Prompt,
前面显示(base)说明已经进入Anaconda的基础环境,然后我们键入

conda create -n pytorch1_11 python=3.7
其中 pytorch1_11是我们创建环境的名称
python=3.7 是我们希望使用的python的版本


我们输入 y 进入下一步。

此时环境已经创建成功,我们可以输入conda activate pytorch1_11激活该环境。

此时已经进入pytorch的环境,我们可以输入pip list检查该环境中有那些包可用。

安装Pytorch

首先进入Pytorch官网
选择与自己版本对应的命令行输入即可
Package:表示包的安装方式,一般window下使用Conda就可以,Linux系统下可以考虑选择pip。
下面是cuda的版本,cuda版本一定要与显卡的型号对应(有些显卡不支持cuda),如果没有显卡或者只有集显,就选择CPU即可。

可以在任务管理器,性能栏查看具体的GPU型号。同时保证显卡驱动版本对应,在安装前最好能把显卡驱动升级到最高。

我们将安装指令复制,输入到刚刚准备好的环境中

一路向下就安装成功了。

最后如果想检验安装是否成功,可以在命令行窗口输入python进入python命令行,再依次输入:

import python
torch.cuda.is_available()

若显示Ture,则证明成功。

安装jupyter notebook

目前我们的环境中还没有jupytet,所以我们需要为刚刚创建好的环境装入jupyter.
在命令行窗口输入:

conda install nb_conda


上面就是我们启动jupyter所需要的一些包,然后我们键入y。

完成安装,这时可以输入jupyter notebook启动jupyter notebook查看安装情况。

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

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

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