要在具体用加载模型代码的文件里面加上
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
才有效果
文章标签 加载 #include i 文章分类 代码人生
要在具体用加载模型代码的文件里面加上
import os
os.environ["CUDA_VISIBLE_DEVICES"] = "1"
才有效果
本文简单介绍了使用Python解决使用GPU的方法,并给出了TensorFlow和PyTorch示例;本文还介绍了TensorFlow GPU的安装版本介绍,以及介绍了安装CUDA的详细教程。
约束训练
举报文章
请选择举报类型
补充说明
0/200
上传截图
格式支持JPEG/PNG/JPG,图片不超过1.9M