当前位置:   article > 正文

Pytorch 安装笔记 可能麻烦但是一定能成功 GPU版本_cu117 torch

cu117 torch

本人最近跑代码YOLOV8需要更新torch版本,看到师弟安装pytorch困难我就想做个笔记大家

网上的各种pip和一键安装对于初学者来讲是行不通的,官网的一键安装也很难实现,当然配置清华源的会很快,不过我要分享的我的安装方法给各位

首先你看到我的时候一定看了很多人了,所以我相信的acananda已经装好了

前提:

1.安装了ananconda

2.配置了环境pytorch

没有完成的先看别人的或者哔哩哔哩 ,我这是笔记,对已安装不成功的人来说好用哟

(分享两个博文:

分别讲了其中之一的安装方法

http://t.csdn.cn/RXGTR

http://t.csdn.cn/EhlSH

第一步:查询

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查看 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/很楠不爱3/article/detail/590282
推荐阅读
相关标签
  

闽ICP备14008679号