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

大家好，又见面了，我是你们的朋友全栈君。如果您正在找激活码,请点击查看最新教程,关注关注公众号 “全栈程序员社区” 获取激活教程,可能之前旧版本教程已经失效.最新Idea2022.1教程亲测有效,一键激活。

Jetbrains全系列IDE使用 1年只要46元 售后保障 童叟无欺

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的虚拟环境

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 不报错则绑定虚拟环境成功