如果显示cuda可用,并且pytorch tensor可正常放到cuda上面去计算,那么就表明整个pytorch+cuda环境搭建成功,后面的for循环主要是为了测试GPU的显存是否能正常被pytorch调用。
windows上报错:Could not find a version that satisfies the requirement torch==0.4.1->安装pytorch0.4.1 :https://blog.csdn.net/weixin_41010198/article/details/103107083
pytorch gpu~ cuda cudacnn安装是否成功的测试代码 2020-03-05 00:02 −... 冰河入梦~ 0 3465 pytorch 指定GPU训练 2019-11-29 11:21 −# 1: torch.cuda.set_device(1) # 2: device = torch.device("cuda:1") # 3:(官方推荐)import os os.environ["CUDA_VISIBLE_DEVICES"] = '1' (同时调...