Pytorch最新安装教程(一步到位)

在这里插入图片描述

  1. 安装anaconda

直接官网安装最新版即可。然后一路next,下图两个选项都点上。
[外链图片转存失败,源站可能有防盗img-niRp src=“C:\Users\92475\AppData\Roaming\Ty-ora\typora-u1erimages\image-202303148262003084.png” referrerpolicy=“no-referrer” alt=“image-C02303:82\4003084”>
ers\92475\AppData\Roaming\Typora\typora-user-images\image-20230318214003084.png)]
接着创建一个pytorch环境,py3.7.2是环境名,python=3.7.2是python版本

创建环境 : conda create -n py3.7.2 python=3.7.2

激活环境 : activate py3.7.2

  1. 切换清华源

2.1 删除之前的镜像源,恢复默认状态

conda config --remove-key channels

2.2 添加清华镜像源

conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main
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/r
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/pro
conda config --add channels http://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/msys2

补充:conda的常用命令

查看所有虚拟环境 conda info --env 在显示的环境中,有*的表示当前所在的环境
创建环境 conda create -n envname python=3.6(指定python版本为3.6)
启动环境 conda activate envnane
关闭环境 conda deactivate envname
删除环境 conda remove -n envname --all
克隆环境 本地已有一个环境AAA,若需要创建一个同样的BBB conda create -n BBB --clone AAA

  1. 安装mamba

mamba是conda的加速神器,可以用来并行下载和安装,大大加快速度,减少失败几率

conda install mamba -c conda-forge
conda install mamba -n base -c conda-forge
  1. 在线安装Pytorch

先查看电脑支持的最大cuda版本,在命令提示符中输入nvidia-smi

[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-ClmA0Ur0-1679147114667)(C:\Users\92475\AppData\Roaming\Typora\typora-user-images\image-20230318212154210.png)]

从官网找到下载指令,cuda版本不能大于电脑支持的最大版本,同时将conda改为mamba

mamba install pytorch torchvision torchaudio pytorch-cuda=11.7 -c pytorch -c nvidia

注意:torch、torchvision与python版本也有关系,所以之前建立的python环境版本也要相符[外链图片转存失败,源站可能有防盗链机制,建议将图片保存下来直接上传(img-eaYs4Ktt-1679147114667)(C:\Users\92475\AppData\Roaming\Typora\typora-user-images\image-20230318212325330.png)]

  1. 验证是否安装成功

进入python环境中:

python

查看torch版本

import torch
print(torch.__version__)

检查Cuda:

torch.cuda.is_available()

显示true,成功结束!

  1. 设置python解释器

如果需要使用本虚拟环境在Pycharm进行项目开发,将设置里的Project Interpreter改为相应Anaconda文件目录下的Pytorch虚拟环境中的python.exe文件即可。(如:D:\Anaconda3\envs\pytorch\python.exe)

  • 1
    点赞
  • 2
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值