- 博客(3)
- 收藏
- 关注
原创 关于AssertionError: Torch not compiled with CUDA enabled解决办法
我使用了conda进行安装,首次安装的CUDA 11.8没有成功,is_available 还是False,最后安装CUDA12.1才成功。安装CUDA12.0 对应的Pytorch版本(我的CUDA12.0被Pytorch1.9.0以上的版本所支持)之后管理员身份进行安装 (安装前先把之前的CUDA卸载干净,在控制面板->程序中卸载)在win+R终端输入nvidia-smi出现下面截图 这里我的显卡是支持CUDA的。首先测试自己电脑是否下载cuda 下面是我电脑的情况,需要下载cuda。
2024-04-18 22:36:27 215
原创 Social LSTM源码阅读
对于每一个序列(20个帧,每帧的人数,每个人的数据 例如20x3x3:拿到x_seq ,_ , d_seq, numPedsList_seq, PedsList_seq通过定义的函数dataloader.convert_proper_array把x_seq变换成适合输入的形式,具体来说只是把PedId-> 0 1 2 数组下标的形式把x_seq的绝对坐标转换成相对坐标 相同的PedId的所有坐标 - 第一次出现的位置。
2024-04-15 19:02:27 270
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人