什么版本的Fastai、torch和torchvision与cuda 10.1、python 3.7兼容?

2024-06-17 12:07:40 发布

您现在位置:Python中文网/ 问答频道 /正文

我为安装fastai创造了一个新环境,fastai依赖于torch&;火炬视觉套装

安装后,我得到:RuntimeError:检测到PyTorch和torchvision是用不同的CUDA版本编译的。PyTorch的CUDA版本为9.2,torchvision的CUDA版本为10.1。请重新安装与PyTorch安装相匹配的torchvision

因此,建议在Cuda10.1&;python 3.7


Tags: 版本环境torchpytorch视觉fastai建议cuda