发布时间：2024-10-25 18:01
两者同时存在时使用gpu
要想使用cpu，在代码里添加如下命令：
matrix1=tf.constant([[3.,3.]])
matrix2=tf.constant([[2.],[2.]])
with tf.Session() as ses:
with tf.device(\"/cpu:0\"):
product=tf.matmul(matrix1,matrix2)
说明一下：
“/cpu:0” The CPU of your machine
“/gpu:0” The GPU of yout machine ,if you have one
如果要换成gpu的话，照着改可以了
或
os.environ[“CUDA_VISIBLE_DEVICES”] = \"-1\"
一行即可