1、cuda

如何查看已安装的cuda版本?

在cmd中输入:nvcc --version

2、torch

torch下载链接:download.pytorch.org/whl/torch_stable.html

在里面根据自己的需要选取合适的torch下载即可,这里我选择的是这个:

torch比较大,大概2个多G

下载完成后,直接把它放入要安装的pycharm项目中,然后在pycharm终端使用:

pip install torch-2.0.0+cu117-cp38-cp38-win_amd64.whl命令安装即可

部分内容参考文章:cuda11.2版本的对应安装的pytorch版本_cuda11.2对应的pytorch-CSDN博客

3、torchvision

torch和torchvision对应的版本参考这篇文章:torch torchvision版本对应关系_torch与torchvision版本对应-CSDN博客

按照上面的torch,本文选择的torchvision版本如下:

下载完成之后,安装方法同torch:

pip install torchvision-0.15.0+cu117-cp38-cp38-win_amd64.whl

Logo

有“AI”的1024 = 2048,欢迎大家加入2048 AI社区

更多推荐