Pytorch从0开始安装

m0_74401015

已于 2022-10-27 13:50:38 修改

阅读量842

点赞数 1

文章标签： pytorch python 深度学习

于 2022-10-23 20:06:19 首次发布

本文链接：https://blog.csdn.net/m0_74401015/article/details/127474548

版权

1安装所需要的工具

1安装cuda，cudnn

CUDA Toolkit Archive | NVIDIA Developer

安装

我选的是Network下载

然后就是一路下一步

查看自己是否安装成功

cuDNN Archive | NVIDIA Developer

下载cudnn需要注册一个账号

登陆后下载

下载好后会有三个文件

找到Cuda的目录

C:\Program Files\NVIDIA GPU Computing Toolkit\CUDA\v11.8

把这三个文件复制进去

anaconda可以帮助我们下载python的库。

2下载安装anaconda

Anaconda | Anaconda Distributionhttps://www.anaconda.com/products/distribution

建议别装D盘

勾选好点击安装

一路Next

点击Finish安装完成

我使用的pycharm来连接anacond

3安装pycharm

Download PyCharm: Python IDE for Professional Developers by JetBrains

我这边安装的是pycharm2022.2.3

写好文件地址点击next

可以按照自己的需要点击，但是最好添加PATH

2开始搭建

1打开我们之前安装的anaconda

conda create -n pytorch python=3.9

创建一个名字为pytorch，python版本为3.9的一个虚拟环境

进入我们的虚拟环境我们安装包就可以从这里install

我们需要使用pycharm来连接anaconda

点击New project

点击Add interpreter来引入我们刚刚创建的虚拟环境

虚拟环境文件一般都在

anaconda/envs/

点击create

pycharm右下角可以看到当前使用的anaconda环境

2下一步安装pytorch，

torch，torchvision，torchaudio

官网有对应的表

Previous PyTorch Versions | PyTorch

按照表上的内容选择

我的cuda版本对应11.3

使用anaconda下载

成功下载

3测试

安装easyocr

www.jaided.ai模型库里下载模型

上代码测试

#@author=m0_74401015

import os
os.environ['KMP_DUPLICATE_LIB_OK'] = 'True'
import re
import sys
import easyocr
import pyautogui
import pygame
import pyperclip
from pygame.locals import *
import string


def _getXY():#显示截取到的屏幕，使用两个点对其切割

    pygame.init()#初始化
    getScreen = pygame.display.Info()
    screen=pygame.display.set_mode((getScreen.current_w,getScreen.current_h),pygame.NOFRAME)#定义窗口
    bacground = pygame.image.load(r"windowsimg.png")#图片
    i=0
    list1=[]
    # list2=[]
    # clist=[list1+list2]
    while True:
        screen.blit(bacground,(0,0))
        for m in pygame.event.get():
            if m.type == MOUSEBUTTONDOWN: #获取鼠标动作
                pos = pygame.mouse.get_pos()
                # print(pos)
                list1.append(pos)
                i=i+1
            if m.type == MOUSEWHEEL:
                return 0 #仅退出此函数返回0
                # sys.exit() #此命令导致图片未分析就结束程序
        pygame.display.update()
        if i == 2:  #获取到两个坐标退出
            img = pyautogui.screenshot(region=(list1[0][0], list1[0][1], list1[1][0] - list1[0][0], list1[1][1] - list1[0][1]))
            # img = pyscreenshot.grab(bbox=(list1[0][0], list1[0][1], list1[1][0] - list1[0][0], list1[1][1] - list1[0][1]))
            img.save('Cutpicture.png')

def get_window():#截取屏幕
    img = pyautogui.screenshot(region=None)
    img.save('windowsimg.png')

def img_SwitchScripts():#切割好的图片转文字并分析

        i = []
        reader = easyocr.Reader(['ch_sim','en'],model_storage_directory=r'./.EasyOCR/model')
        readimg = reader.readtext(r'Cutpicture.png')
        for r in readimg:
            i.append(r)
            # i+=r
        l=tuple(i)
        for li in l:
            li=re.findall(r"['](.*?)[']",str(l))
        lii="".join(li)
        # print(text)
        pyperclip.copy(lii)
        # pyperclip.paste()

if __name__ == '__main__':#入口

    get_window()
    _getXY()
    img_SwitchScripts()