PyTorch安装教程(最简单方法)

注意:在安装之前确定你要安装的pytorch版本,然后在PyTorch和CUDA版本对应关系中查找对应版本,再进入正式安装环节

之前使用anaconda进行安装,下载速度实在太慢,安装几次都失败了。后听我朋友建议采用pip进行安装,直接成功!!!
在这里插入图片描述

一、根据之前你查看的对应版本,安装对应的 cuda 和 cudnn(我当时安装是参考的1-cuda和cudnn安装2-cuda和cudnn安装

二、cuda 和 cudnn都安装成功后,采用pip安装PyTorch

1、查看 cudnn 和 cuda 版本

#查看cudnn 版本
nvcc -V

#查看cuda版本
set cuda

在这里插入图片描述

2、进入pytorch官网:https://pytorch.org/,下载对应版本(PyTorch和CUDA版本对应关系

  • 网页下拉,即可看到下图,可安装的CUDA版本,给出了安装命令。如果没有找到对应版本,或者该版本没有安装命令,则进行下一步
    在这里插入图片描述
  • 查看所有版本
    在这里插入图片描述
    跳转进入下面页面
    在这里插入图片描述
    下滑找到需要的pytorch版本
    在这里插入图片描述

2、打开PyCharm软件,打开终端,粘贴安装代码
在这里插入图片描述

等待安装,我当时安装大概十分钟左右(用conda安装好几个小时还安装失败!!)
3、查看安装是否成功:

  • 使用pip list检查
    在这里插入图片描述
  • 不确定的话,还可以进行代码测试
    import torch
    print(torch.__version__)
    print(torch.cuda.is_available())
    print(torch.version.cuda)

在这里插入图片描述

这样就成功啦!!

这是小新学习的小记录,有不对的地方希望UU们批评指正
在这里插入图片描述

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
青葱年少的头像青葱年少普通用户
上一篇 2023年3月5日
下一篇 2023年3月5日

相关推荐