当前位置:   article > 正文

学习笔记——从零开始安装pytorch,并在pycharm中使用_pycharm怎么用pytorch

pycharm怎么用pytorch

1.Anaconda安装

参考博主这篇文章

https://blog.csdn.net/weixin_51130521/article/details/119414480 

2,启动Anaconda Prompt 创建虚拟“房间”

在这里插入图片描述

通过conda创建一个名为:pytorch的虚拟“房间”,可以方便未来对不同版本的PyTorch进行管理,具体方法如下:

创建房间

conda create -n pytorch python=3.6

 在这里插入图片描述

在这里插入图片描述

  查看创建的“房间”

conda info --envs

3.安装PyTorch

激活pytorch房间

conda activate pytorch

在这里插入图片描述

当前面从(base)变为(pytorch)时表示此时已经切换到pytorch房间,随后正式进入安装pytorch环节。

通过pytorch官网,点击Get Started

在这里插入图片描述

随后根据自己计算机环境选择相应的Pytorch版本。

在这里插入图片描述

其中CUDA时表示当你的计算机有GPU时,随后将下图的代码粘贴进Anaconda Prompt。 

在这里插入图片描述

如图所示:

在这里插入图片描述 回车安装所需要的第三方包,其中pytorch 1.6.0、torchvision 0.7.0等由于缺乏镜像资源,因此需要我们手动下载。后面会具体详述

在这里插入图片描述

这里需要注意的是一定要在(pytorch)这个虚拟房间下,运行安装。
其他无法获得镜像安装的可以参照上述离线方法安装。 

4 安装PyCharm

参考博主这篇文章https://blog.csdn.net/weixin_51130521/article/details/119951176

5在pycharm中添加pytorch房间的解释器。

 在这里插入图片描述

点击设置,在python解释器中,点击Add
在这里插入图片描述

选择Conda Environment,选择已经存在的环境。并选择pytorch房间的pytorch.exe解释器。
在这里插入图片描述

完成后,点击OK即可。随后返回主界面,但是需要等一到两分钟,用于初始化 

验证一下

  1. import torch
  2. import numpy as np
  3. arr=np.ones((3,3))
  4. print("arr的数据类型为:"+str(arr.dtype))
  5. t=torch.tensor(arr)
  6. print(t)

在这里插入图片描述

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/不正经/article/detail/331661
推荐阅读
相关标签
  

闽ICP备14008679号