markchalse的博客

工欲善其事，必先利其器显卡驱动版本和cuda版本？今天两台电脑训练时发现速度比平时慢了，以为是网络的变大导致。但nvidia-smi发现显存占用少，gpu速度占用1% ，同时用gnome-system-monitor 发现cpu全部100%，难道pytorch用的cpu训练？可是pytorch装得时候没问题，训练时也没问题。这时发现自己代码里的device = torch.device("cuda:0" if torch.cuda.is_available() else "cpu")让我后背一凉..

硬件信息cpu Intel(R) Core(TM) i7-10700 CPU @ 2.90GHz显卡 GeForce RTX 3090网卡 Ethernet Connection (17) I219-V内存 62GiB System memory----------------------------------------------------------------------------------换源sudo gedit /etc/apt/sources.li...