Anaconda下载simpleITK包
根据你的需求确定是下载cpu版本还是gpu版本
去pytorch官网
根据自己的需要选择是什么系统(linux/win),用什么工具(conda/pip),什么语言(python/java/c++),什么版本(cpu/gpu)
特别的,如果是下gpu版本的话,看一下自己的cuda和cudnn下好了没,不然下载了pytorch包也用不了
win+R进入cmd
输入nvidia-smi查看cuda版本
这样就知道cuda版本了,至于没有下载好cuda和cudnn的自己去百度吧,有空姐再写博客
测试一下能不能用~
import torch
print