安装之前,先配好环境
- 即装CUDA和CUDANN 具体是什么,可以去搜一下
- 这里 我参考了这篇文章都看看
- Pytorch官网链接
- 这里进行下补充,我下载的是cuda11.8,在官网中,首先点击 install previous version
搜索11.8,可以看到这条命令,在Anaconda Prompt中输入该条命令安装pytorch gpu版本
同时记录下自己的torch和torchvision版本 2.0.0 和 0.15.0
安装方法同上文链接中的方法一样
我对python版本为3.8 系统为windows11,则可以选择这个
测试方法同上文。