深度学习常用pytorch框架,安装之前需要确保已经配置好显卡驱动和CUDA,具体教程后面再补充。
1.torch安装:
进入pytorch官网,选择对应版本进行安装,复制粘贴红框中的命令,进入prompt安装就可以了,效果如下:
2.安装验证:
验证安装版本和GPU是否可用:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
out:
1.10.2
True
深度学习常用pytorch框架,安装之前需要确保已经配置好显卡驱动和CUDA,具体教程后面再补充。
1.torch安装:
进入pytorch官网,选择对应版本进行安装,复制粘贴红框中的命令,进入prompt安装就可以了,效果如下:
2.安装验证:
验证安装版本和GPU是否可用:
import torch
print(torch.__version__)
print(torch.cuda.is_available())
out:
1.10.2
True