在Anaconda Jupyter Notebook中如何安装Pytorch |
您所在的位置:网站首页 › jupyter配置pytorch环境 › 在Anaconda Jupyter Notebook中如何安装Pytorch |
一、安装pytorch环境 1、进入anaconda的默认环境base 两种方法:一种是在anaconda prompt中,直接点开就可以。 代码:conda create -n 环境名 python=版本 解释:conda是指调用conda包,create是指创建屋子,-n是指后面的名字是屋子的名字,环境名是指屋子的名字,python=版本是指创建的屋子是python的某个版本的 所以其实我们需要查看一下现在已经有哪些环境库:conda info -e 然后查看一下python的版本:python -V ok,那么用conda create -n pytorch python=3.8来创建一个pytorch环境。 4、查看电脑上CUDA版本: cmd中输入“nvidia-smi” 5、pytorch安装 去pytorch官网:https://pytorch.org/ ,选择对应的版本,复制下载命令 6、验证是否安装成功 (1)、进入pytorch环境: (2)、输入python (3)、输入import torch 没有报错说明安装成功。 (4)、输入torch.cuda.is_available() ,如果结果是True则说明可以使用GPU,False就意味着只能使用CPU。 来源:https://blog.csdn.net/weixin_47038938/article/details/115470086 |
今日新闻 |
推荐新闻 |
CopyRight 2018-2019 办公设备维修网 版权所有 豫ICP备15022753号-3 |