配置YOLOV5 小记(二)配置pytorch(CPU/GPU)

—————————————————————————————————————————————

这一章

1.配置pytorch,主要源于 利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装—免额外安装CUDA和cudnn(适合小白的保姆级教学)

2.对 总结配置pytorch GPU环境的两种方法中的:离线配置环境 进行总结

—————————————————————————————————————————————

Anaconda安装及 环境配置

不同Anaconda对应到的python版本不同,而且后期的 没有VScode(vscode去官网下也很方便)
Anaconda各历史版本下载.

Anaconda安装及 环境配置这个非常靠谱:
利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装—免额外安装CUDA和cudnn(适合小白的保姆级教学).

装pytorch需要在Anaconda Prompt进行
在这里插入图片描述
先提供几条常用语句
进入python环境(同时可见python版本)

python

退出python环境

exit()

卸载已经下载的(torch)

pip uninstall torch

换源

pip config set global.index-url https://pypi.tuna.tsinghua.edu.cn/simple

阿里云 http://mirrors.aliyun.com/pypi/simple/
清华大学 https://pypi.tuna.tsinghua.edu.cn/simple/
中国科技大学 https://pypi.mirrors.ustc.edu.cn/simple/
豆瓣(douban) http://pypi.douban.com/simple/
中国科学技术大学 http://pypi.mirrors.ustc.edu.cn/simple/

—————————————————————————————————————————————
CPU装pytorch只需要一句:
不指定版本

pip install torch torchvision torchaudio

安装指定torch版本:

pip install torch==1.7.1 -i https://pypi.tuna.tsinghua.edu.cn/simple

同理 还有torchvision torchaudio

前面版本号,后面是源。下载慢一般都要换源。

检验是否成功

python
import torch
print(torch.__version__)

print(torch.__version__) 可以打印 当前版本号+CPU

—————————————————————————————————————————————

GPU装pytorch:

两种方法: 总结配置pytorch GPU环境的两种方法.
离线下载的那个最香

(重点)简要概括离线的方法:把包下载好,在用代码本地安装。速度非常快。

0.1 打开pytorch页面
如果页面.给的版本下载后 检验(python环境下 import torch)发现安装不成功,则按下图进入历史版本寻找

在这里插入图片描述
我用的torch是1.7.1 cuda 11.0
这里的CUDA 小于 电脑的cuda(更新过驱动后)就行
更新方法 见我的上一篇.
在这里插入图片描述
查看自己得cuda版本号方法:

nvcc --version

注意 torch torchvision torchaudio版本号是成套的,不能随意修改哦在这里插入图片描述
0.2开始离线下载
确定要下载的torch torchvision torchaudio版本后进入
torch torchvision torchaudio各版本
用ctrl+F搜索
注意不要 下成CPU/开头的,(踩过坑)
在这里插入图片描述

cu110就是CUDA11.0
%2B是就“+”的意思
cp39就是 python3.9的意思
点击以后 网页下载比较慢。右键 复制下载链接,然后打开迅雷,点新建,十几秒解决战斗。
在这里插入图片描述
在这里插入图片描述
在这里插入图片描述
按上面的操作再下载 torchvision torchaudio。
将下载好的3个文件 放入C:\Users\Administrator
在这里插入图片描述
打开Anaconda Prompt进行
1.查看版本

conda --version

2.查看虚拟环境

 conda info -e

3.创建虚拟环境
因为anaconda自带python,就按它的py版本来

conda create -n pytorch python=3.9.7

4.创建之后激活

activate pytorch

5.在pytorch虚拟环境下
1)键入

dir

出来一大堆。
2)再键入pip install torch然后按tab建 补充完整

(速度不知道比 在线pip 快多少倍)

3)键入pip install torchaudio 然后按tab建 补充完整

4)键入pip install torchvision然后按tab建 补充完整

6.(激活pytorch后)再进入python环境下对pytorch 验证

python
import torch
print(torch.__version__)

在这里插入图片描述
如果是在(base)环境里直接import torch 会这样。(我以为坏了 又重装了一遍)
在这里插入图片描述
后面还需要 将在pycharm里进行“环境 绑定”
跟着利用Anaconda安装pytorch和paddle深度学习环境+pycharm安装—免额外安装CUDA和cudnn(适合小白的保姆级教学).第五步操作就行。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包

打赏作者

是孑然呀

你的鼓励将是我创作的最大动力

¥1 ¥2 ¥4 ¥6 ¥10 ¥20
扫码支付:¥1
获取中
扫码支付

您的余额不足,请更换扫码支付或充值

打赏作者

实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

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

余额充值