pytorch安装以及pycharm使用pytorch包（windows10）

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

1. 安装anaconda

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

为anaconda配置环境变量

huanjing

环境变量设置

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

在这里插入图片描述

输入conda显示如下

在这里插入图片描述

则证明anaconda安装成功

2. 创建虚拟环境

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

conda create -n pytorch38 python=3.8

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

位置

3. 安装pytorch

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

配置

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

conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch

等待安装完成后，输入

import torch
x = torch.rand(5, 3)
print(x)

输出

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]])

输入

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