pytorch
我比青菜高
这个作者很懒,什么都没留下…
展开
-
python(杂)不定时更新
cuda上面的变量类型只能是tensor,不能是其他原创 2020-07-10 16:41:52 · 162 阅读 · 0 评论 -
pytorch,令代码在指定gpu上运行
所用服务器有四个卡0、1、2、3,想指定代码在2卡上运行。不知道为啥,用import osos.environ['CUDA_VISIBLE_DEVICES'] = '2'和export CUDA_VISIBLE_DEVICES= 2最后用nvidia-smi看代码都在0卡上跑。。最终解决:import torchtorch.cuda.set_device(1)我刚开始改成只用torch,会提示我设备号不存在,加上os之后可以执行,后来又试了几次删掉os,也可以执行???费解.原创 2020-08-13 21:34:10 · 972 阅读 · 0 评论