Pytorch安装教程

  • Post author:
  • Post category:其他


Pytorch安装教程(使用版本CUDA == 10.2)


前言


1.安装Anaconda


2.查看CUDA版本


3.安装Pytorch


4.测试是否安装成功


前言

在测试中一直提示False,后来又看了下其他教程,发现其他博主都是针对CUDA>10.2版本进行安装,后来参考了好多文章,终于找到如何解决这个问题,现在就把整体的安装教程写出来供大家参考。


安装Anaconda

理论知识就不给大家整理了,直接开始安装教程!!!

Anaconda下载链接:



官方下载链接:



Anaconda | The World’s Most Popular Data Science Platform

可能下载速度比较慢,下面给出清华大学开源镜像下载地址



清华源下载地址:



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

往下翻,在清华大学开源软件镜像站中选取较新的的Anaconda安装包

下载完成后双击,开始安装Anaconda

点击I Agree ,同意安装协议

选择Just Me,点击Next进行下一步

选择安装目录,点击Next进行下一步

勾选下面两个选项,进行安装

进入Anaconda Prompt(Anaconda)中,将conda下载地址更改为清华源下载地址:

#  将下载通道地址修改为清华源下载地址
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/main/
conda config --set show_channel_urls yes

#  更新conda
conda update conda
#  更新所有已安装conda包
conda upgrade --all

进入C盘目录C:\Users\Administrator,找到.condarc文件,选择记事本格式打开


2.查看CUDA版本


3.安装Pytorch

创建pytorch环境,


PyTorch为环境名

conda create -n PyTorch python=3.8

查看已创建完成的环境

conda info --envs

进入到环境名为PyTorch的环境中

conda activate PyTorch

开始安装pytorch环境

pip install torch==1.10.0+cu102 torchvision==0.11.0+cu102 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html


4.测试是否安装成功

python

import torch

print(torch.__version__) # pytorch版本

print(torch.version.cuda) # cuda版本

print(torch.cuda.is_available()) # 查看cuda是否可用



版权声明:本文为u011669976原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。