window10配置PyTorch相对比较简单,只需要进入如下链接:
https://pytorch.org/get-started/locally/
然后选择自己需要的配置即可,我的是:
然后把红色部分复制下来就可以进行安装了。
安装成功后可以输入指令测试:
import torch
torch.cuda.is_available()
输出True
就表示安装成功了。
window10配置PyTorch相对比较简单,只需要进入如下链接:
https://pytorch.org/get-started/locally/
然后选择自己需要的配置即可,我的是:
然后把红色部分复制下来就可以进行安装了。
安装成功后可以输入指令测试:
import torch
torch.cuda.is_available()
输出True
就表示安装成功了。