使用GPU(tensorflow)在代码开头加入
import tensorflow as tf
physical_devices = tf.config.list_physical_devices('GPU')
tf.config.experimental.set_memory_growth(physical_devices[0], True)
print("Before running the script:")
print(tf.test.is_gpu_available())
查看GPU使用情况
静态:nvidia-smi
动态:watch -n 10 nvidia-smi。10表示10s间隔
看不到GPU使用情况,fuser -v /dev/nvidia*查看使用gpu的进程号
查找虚拟环境所在目录(设置python解释器)
进入虚拟环境后,命令行输入:where python
创建:conda create -n name(name 是自己取的名字)
安装包:pip install -r requirements.txt (进入req的文件夹)
查看现有虚拟环境: conda info --envs (参考这篇文章)https://blog.csdn.net/chengyq116/article/details/109965489
错误提示:
ModuleNotFoundError: No module named sklearn
解决:pip install scikit-learn
module‘object is not callable
module'object is not callable
错误代码:optimizer = adam_v2(learning_rate=0.0001, beta_1=0.5)
解决: optimizer = adam_v2.Adam(learning_rate=0.0001, beta_1=0.5)
参考:我如何解决这个错误"'module‘object is not callable“ - 问答 - 腾讯云开发者社区-腾讯云
if __name__ != '__mp_main__'和 if __name__ != '__main__'
__name__
是Python中一个特殊的内置变量,当一个Python文件作为主程序运行时,它的值被设置为__main__
。当脚本作为模块导入到另一个脚本中时,__name__
的值将被设置为模块的名称,而不是__main__
。