设置cuda——launch——blocking import os os.environ['CUDA_LAUNCH_BLOCKING'] = "1" 好文要顶 关注我 收藏该文 微信分享 祥瑞哈哈哈 粉丝- 2 关注- 0 +加关注 0 0 升级成为会员 « 上一篇: 保存图片 » 下一篇: malab把single数据保存为tif ...
此外,还在执行slurm作业的命中中声明了其他的指定,我使用的指令如下: sbatch -N 1 -n 4 --nodelist=comput2 sleep.sh 1. 对于CUDA_LAUNCH_BLOCKING=1问题 用pip卸载原来pip安装的torch,并用conda安装指定的torch版本和对应的cuda版本(用conda安装),我用到的指令为: conda install pytorch==1.10.1 torchvision=...