赞
踩
下图确认是否有cuda
- import torch
- print(torch.__version__)
- print(torch.cuda.is_available())
若为false,说明我们需要安装cuda
首先需要在命令行输入nvidia-smi,查看个人电脑可使用的cuda版本
我在命令提示符中直接下载的torch为2.2版本,根据适配的版本,我最开始选择的是cuda12.1版本,但在最后安装的时候出现安装失败的情况(此处忘记截图了)
(此时注意可以在电脑控制面板-程序-程序与功能-在搜索栏输入NVIDIA卸载掉已下载的部分,再下载其他版本)
经过网上的查阅,最终选择下载cuda11.8版本,这个版本可以正常安装,再通过pytorch官网找到对应的pytorch版本的命令,
下图是pytorch下载官网中的选择
-----------------------------------------------------------------------------------------------------------------------------(补充)安装成功后,jupyter查看torch版本仍为2.2.0,即使是在命令行中卸载pytorch也没有用,此时直接在文件夹搜索栏,搜索pytorch,找到2.2.0版本文件,删除即可
------------------------------------------------------------------------------------------------------------------------------
重启jupyter,又出现jupyter挂掉的情况,网上说输入
import os
os.environ["KMP_DUPLICATE_LIB_OK"]="TRUE"
可以解决此问题,但问题依旧存在,最后是重启大法成功解决(个人认为是清除了加载缓存,解决重复加载的问题,因此可以恢复运行)
Copyright © 2003-2013 www.wpsshop.cn 版权所有,并保留所有权利。