赞
踩
本人最近跑代码YOLOV8需要更新torch版本,看到师弟安装pytorch困难我就想做个笔记大家
网上的各种pip和一键安装对于初学者来讲是行不通的,官网的一键安装也很难实现,当然配置清华源的会很快,不过我要分享的我的安装方法给各位
首先你看到我的时候一定看了很多人了,所以我相信的acananda已经装好了
前提:
1.安装了ananconda
2.配置了环境pytorch
没有完成的先看别人的或者哔哩哔哩 ,我这是笔记,对已安装不成功的人来说好用哟
(分享两个博文:
分别讲了其中之一的安装方法
第一步:查询
1.查看自己的python版本
2.cuda
我的CUDA Version: 12.1 所以高版本的cuda117、118 都可以 但是我装117 别太高
3.torch与torchvision版本
目前主流3.8、3.9 、3.10 用户最多,cuda目前官网的提示时117、118
我自己版本时python3.9 对应117
torch与torchvision版本对应查询 连接:http://t.csdn.cn/3M0SS
第二步 下载
参考官网的 连接你技能看出来我要下载python3.9 cuda为117 64位电脑的,也可以自己查
下面是下载连接
https://download.pytorch.org/whl/torch_stable.html
然后找到版本的torch 和torchvision 这两个都要安装你才是真的安装成功啦
参考:
pytoch官网:https://pytorch.org
Downloading https://download.pytorch.org/whl/cu117/torch-2.0.1%2Bcu117-cp39-cp39-win_amd64.whl (2343.7 MB)
第三步安装
1.把下载好的两个文件放到D:\Anaconda3\Lib\site-packages 路径下 就是安装位置下的配置安装包位置
2. 打开ananconda的prompt 页面
3.输入激活环境,开头变了就可以啦
conda activate pytorch
4.定位到文件夹
输入:
cd /d D:\Anaconda3\Lib\site-packages
如果定位不成功自己百度如何去cd 定位到文件位置
5.安装
分别输入pip install “你下载的包名字.whl"
pip install "torch-2.0.1+cu117-cp39-cp39-win_amd64.whl"
pip install "torchvision-0.15.1+cu117-cp39-cp39-win_amd64.whl"
就可以啦,他会自己卸载你之前的错误版本并安装成功,别忘了.whl 哟
(我的版本 4060显卡
cu117
torch 2.0.1
torchvision 0.15.1)
6.最后检查: 其他人的博文都会说,难得配图了
1.1方式1输入:
python 就可以运行代码了
import torch
print(torch.cuda.is_available())
看是否为ture
2.2方式2 输入 conda list查看
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。