一.查看自己电脑的配置
首先查看自己电脑的cuda的版本,Win+R,敲入cmd打开终端
输入nvidia–smi,查看自己电脑的显卡等配置
这里要说明一下关于这个CUDA,它具有向后兼容性,这意味着支持较低版本的 CUDA 的应用程序通常也可以在较高版本的 CUDA 上运行
二.安装pytorch
在安装pytorch和相关包的时候,我建议是去官方网站去找你所需要的版本;贴一个pytorch官方网址https://pytorch.org/ 你可以在这找你所需要的版本 或者是这个里面自己找你所需要的版本https://download.pytorch.org/whl/torch/
第一个方法:https://pytorch.org/
如果在刚才的地方找不到你所需要的版本就需要到之前的板块去找
举个例子,我的电脑的cuda是12.1的版本,但是现在我要复现的项目所需要的torch版本是1.13.0,那么我就可以选择cuda116的或者是cuda117的,下载的时候会给出对应的conda命令和pip的命令,还有不同系统的命令也会有不同,这点要注意!如果你所使用的是pip下载的,你也可以直接到网址把对应的torch下载到本地,然后再导入到你的项目里,直接访问url就能够下载。
第二个方法:https://download.pytorch.org/whl/torch/
Ctrl+F根据自己所需和文件结构直接搜索
pip install 文件路径/文件名.whl命令可以导入下载的文件
加号前面的是torch的版本,后面cpu表示是cpu的版本,cp37是对应的python的版本是3.7.x的
三.安装torch对应的包
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。