pytorch安装以及pycharm使用pytorch包(windows10)

2022-09-27 15:40:12 浏览数 (1)

大家好,又见面了,我是你们的朋友全栈君。

1. 安装anaconda

https://www.anaconda.com/products/individual#windows 在该网站上下载anaconda的包,然后使用默认设置安装即可。

为anaconda配置环境变量

anaconda和miniconda同为conda官方推荐的conda工具。anaconda和conda的关系,就像github和git的关系,anaconda不光包含了conda命令行软件,还包含了多如牛毛的预装python包,以及令人瞠目结舌的黑科技全家桶。 win R打开命令行。输入python显示如下

输入conda显示如下

则证明anaconda安装成功

2. 创建虚拟环境

因为深度学习不同网络的环境不同,所以pytorch通过支持虚拟环境以应对不同的python等版本需求。 用以下命令创建名为pytorch38的虚拟环境

代码语言:javascript复制
conda create -n pytorch38 python=3.8

该虚拟环境所在位置为anaconda安装目录下envs文件夹下

3. 安装pytorch

在跳出的cmd框中根据你选择的配置输入命令安装,配置选择在https://pytorch.org/get-started/locally/中

按照我选择的安装需求我在命令行输入

代码语言:javascript复制
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

等待安装完成后,输入

代码语言:javascript复制
import torch
x = torch.rand(5, 3)
print(x)

输出

代码语言:javascript复制
tensor([[0.3380, 0.3845, 0.3217],
        [0.8337, 0.9050, 0.2650],
        [0.2979, 0.7141, 0.9069],
        [0.1449, 0.1132, 0.1375],
        [0.4675, 0.3947, 0.1426]])

输入

代码语言:javascript复制
import torch
torch.cuda.is_available()

返回true,则安装正常。 如果报错,可能是并未安装VS的原因,可以根据命令行提示中所给网址下载安装网址(只是一个包的安装,并不复杂,下载是个安装器直接安装即可)

4. pycharm绑定虚拟环境

首先你需要安装pycharm,官网地址为https://www.jetbrains.com/pycharm/download/#section=windows,下载后默认设置安装即可。 新建工程

绑定虚拟环境

5. 检查安装结果

输入import torch 不报错则绑定虚拟环境成功

发布者:全栈程序员栈长,转载请注明出处:https://javaforall.cn/175310.html原文链接:https://javaforall.cn

0 人点赞