深度学习环境配置

先执行  conda create -n （自己的名字） python=3.7  创建虚拟环境

activate （自己的名字） 激活自己的虚拟环境

 测试的test.py

import torch
import time

a = torch.randn(400,600,600).to("cuda:0")

b = torch.randn(400,600,600).to("cuda:0")

start_time = time.time()

for i in range(1,1000):
    c = a*b

end_time = time.time()

print("CUDA time: ",(end_time-start_time))


a = torch.randn(400,600,600)

b = torch.randn(400,600,600)

start_time = time.time()

for i in range(1,1000):
    c = a*b

end_time = time.time()


print("CPU time: ",(end_time-start_time))

运行结果