欢迎光临散文网 会员登陆 & 注册

pytorch环境搭建(离线方式)

2023-03-20 15:05 作者:吠落  | 我要投稿

前言

    网上搜到的环境安装教程很多,离线安装的方式是我认为成功率最高,速度最快的方式。我把我搜到的教程总结出来。

  1. 准备

    (1)安装python(建议版本3.7-3.8)

    (2)安装Anaconda

    (3)安装pycharm

    以上三步直接从官网下载,c盘够用的话安装的时候无脑下一步就行,不然就自己分配位置。

  2. 使用conda创建环境

    (1)查看显卡驱动

    在命令行中(win+r 输入cmd打开)输入nvidia-smi,查看CUDA版本。

这里我的cuda版本是11.6,之后我们需要下载对应版本的资源包,如果不显示这个界面,之后下载的资源包就选择cpu版本

        (2)conda创建环境

  • 打开开始菜单,进入anaconda prompt(右键->更多->以管理员身份运行)。

注意看对名字
  • 输入命令:conda create -n name python=xx   (name:你为这个环境命的名 ,xx:你的python版本)

这里我取的环境名是mypython,我的python版本是3.7.2(与自己的python版本对应)
  • 之后会有一个选择页面,输入y表示yes

  • 成功后如下,输入conda activate + 你建的环境名,激活环境

 3.在环境中安装离线资源包

    (1)下载3个资源包:torch,torchvision,torchaudio 

        网址:  https://download.pytorch.org/whl/torch_stable.html

  • 进入后页面如下

cpu开头的是CPU版本,cu开头的是GPU版本。“cp” 后面的数字代表 python 版本。“win_amd64”表示适配的操作系统。
  • 例如:我的配置是cuda11.6与python3.7,所以选择:

  • 之后继续下载torchvision和torchaudio ,注意版本对应关系

根据torch的版本下载torchvision
根据torch版本下载torchaudio
  • 最后得到:


可以把这三个文件放到一个文件夹里,方便查找

    (2)安装资源包

  •  安装前,先进行换源

    打开命令行(win + r  cmd),输入命令:

    pip config set global.index-url https://mirrors.aliyun.com/pypi/simple/

这一步的操作会将pip的下载源更换到这个这个网址,这里推荐用阿里云的镜像。
  • 回到之前建的环境下,输入命令:pip install + 资源包的位置

注意这里要在环境中安装,最前面的()显示的是环境名,如果中途关掉了重新进入要重新激活环境:activate + 环境名
  • 全部安装完成

要把之前下载的3个包全装完
  • 检查安装情况。进入python,输入import torch,输入print(torch.__version__)   (这里前后各有两个下划线)

成功检测版本信息,大功告成

4.在pycharm中使用已搭建的环境

  • 打开pycharm


不习惯英文界面可以先下载汉化包,重启生效
  • 新建项目


注意这里不要选到Conda环境
搭建的环境都存在envs文件夹里,找到要使用的环境里的python解释器
确定后返回,点击创建



pytorch环境搭建(离线方式)的评论 (共 条)

分享到微博请遵守国家法律