地球上最后一个直男

摘要： 第一步！指定gpu import osos.environ["CUDA_VISIBLE_DEVICES"] = '0' 第二步！ 对于每一个要踹到gpu去的Tensor或者model x 使用x = x.cuda()就ok了 嘤嘤嘤 阅读全文

摘要： 第一步！启动需要补全的虚拟环境 conda activate pytorch 第二步！安装这玩意儿 pip install jupyter_contrib_nbextensions 第三步！ jupyter contrib nbextension install --user 第四步！安装这玩意儿 阅读全文