安装cuda 、torch、torchvision
整体思路为 先根据 nvidia-smi 指令 看看自己的cuda版本,然后安装对应的cuda驱动 对应关系表如下:
接着,根据安装的cuda以及python版本,安装对应的torch,cuda、python与torch对应版本号如下:
然后就可以安装torchvision了:
原文地址:https://blog.csdn.net/m0_75015083/article/details/144324506
免责声明:本站文章内容转载自网络资源,如本站内容侵犯了原著者的合法权益,可联系本站删除。更多内容请关注自学内容网(zxcms.com)!