Windows下pycharm十anaconda十pytorch训练数据集环境配置
环境配置过程中遇到了很多问题,记录一下。主要问题在于torch和torchvision的安装1、cuda和pytorch版本要对应,torch和torch vision版本要对应,对应关系为2、一个电脑可以配置不同版本的cuda,记得添加环境变量,用哪个cuda时将对应版本环境变量路径置顶,cuda和cudnn下载及安装配置如下:https://blog.csdn.net/m0_37605642/article/details/98854753?utm_source=app3、安装torch时
原创
2021-04-22 14:27:35 ·
149 阅读 ·
2 评论