Step 1
打开官网:
https://pytorch.org/get-started/locally/
进行选择对应版本:
复制图中命令执行。
Step 2
验证是否安装成功。
执行:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)
输出:
Step 1
打开官网:
https://pytorch.org/get-started/locally/
进行选择对应版本:
复制图中命令执行。
Step 2
验证是否安装成功。
执行:
import torch
print(torch.cuda.is_available())
print(torch.cuda.device_count())
print(torch.version.cuda)
输出:
版权声明:
本网仅为发布的内容提供存储空间,不对发表、转载的内容提供任何形式的保证。凡本网注明“来源:XXX网络”的作品,均转载自其它媒体,著作权归作者所有,商业转载请联系作者获得授权,非商业转载请注明出处。
我们尊重并感谢每一位作者,均已注明文章来源和作者。如因作品内容、版权或其它问题,请及时与我们联系,联系邮箱:809451989@qq.com,投稿邮箱:809451989@qq.com