Pytorch安装教程

CV小菜鸡678

已于 2023-11-18 16:15:07 修改

阅读量141

点赞数 2

文章标签： pytorch 人工智能 python

于 2023-11-18 13:49:02 首次发布

本文链接：https://blog.csdn.net/weixin_72356313/article/details/134477045

版权

Pytorch安装教程

说明：此教程以windows系统为例

1.Anaconda

1)下载Anaconda

网址：Index of /anaconda/archive/ | 清华大学开源软件镜像站 | Tsinghua Open Source Mirror

2）安装

选择安装路径：

不要安装在系统盘（一般C盘），在其他盘新建一个文件夹，注意整个路径不要使用中文或者空格、特殊字符。选择对应安装目录，点next。

要是不勾上面的第二个框，就要手动配置环境：

步骤：此电脑----->属性----->高级系统设置----->环境变量----->path----->编辑----->新建

3）检查是否安装完成

cmd,回车：

输入conda查看是否安装成功

输入conda -V命令可查看当前Anaconda版本

输入python --version命令查看是否可以使用python版本

4）更改下载源

打开Anoconda Prompt:

一行一行输入以下命令：

#添加国内地址：

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/conda-forge/

#查看通道地址：

conda config --show channels

注：如果要删除自定义源更换回conda的默认源，直接删除channels即可，命令如下：

conda config --remove-key channels

2.Cuda（没有英伟达显卡可跳过）

1）查看自己的显卡：

在任务栏右键–>任务管理器–>性能

2）：查看电脑cuda版本：

nvidia-smi

3)：查看相对应的版本

[Pytorch 各个GPU版本CUDA和cuDNN对应版本](https://so.csdn.net/so/search?spm=1001.2101.3001.4498&q=Pytorch 各个GPU版本CUDA和cuDNN对应版本&t=&u=)

pytorch对应版本安装

CUDA和cuDNN各版本下载及版本对应关系

4)：下载cuda和cudnn

CUDA安装教程

3.Pytorch

#新建一个新环境：(newPytorch为环境名,自行设置)

conda create -n newPytorch python=3.x(x为对应版本号.如3.10)

#激活环境：

conda activate newPytorch

1)去pytorch官网

https://pytorch.org/

CPU（没有英伟达显卡）

注意要把后面的-c pytorch去掉，不然还是使用的默认源下载。

conda install pytorch torchvision torchaudio cpuonly

GPU

conda install pytorch torchvision torchaudio pytorch-cuda=12.1 -c pytorch -c nvidia
(安装报错可以选择pip进行安装)

#检查是否安装成功

python
import torch
torch.__version__

不报错则安装成功

4.Pycharm

1）下载安装

去官网：https://www.jetbrains.com/pycharm/download/?section=windows，点击下载

下载好后点击安装

自选路径

勾选，直到最后，安装完毕

2）配置环境

打开PyCharm，点击New Project（新建一个项目），选择项目所在地址和解释器

完成