Python3.9安装torch、mvcc

粉色系

已于 2023-03-19 18:09:20 修改

阅读量3.5k

点赞数

文章标签： python pytorch 深度学习

于 2023-03-19 15:36:02 首次发布

本文链接：https://blog.csdn.net/qq_39898555/article/details/129651388

版权

本文介绍了如何在Python3.9环境下安装PyTorch、torchvision、torchaudio和mmcv-full，并针对torch.cuda.is_available()返回false的问题提供了卸载重装的解决步骤。推荐在cmd中通过whl文件直接下载，或在Anaconda的虚拟环境中直接复制安装文件。同时，文章还给出了安装mmcv-full的命令。

摘要由CSDN通过智能技术生成

Python3.9安装torch、mvcc对应版本清单：
torchvision-0.14.0+cu117-cp39-cp39-win_amd64.whl
torch-1.13.0+cu117-cp39-cp39-win_amd64.whl
torchaudio-0.13.0+cu117-cp39-cp39-win_amd64.whl
mmcv-full==1.7.0

下面安装教程适用于解决torch.cuda.is_available()返回false解决办法：重装torch
打开cmd输入

python
import torch
print(torch.__version__)

运行结果：1.7.1+cpu
呵呵，恭喜你中招了！我也是在参考https://www.bilibili.com/video/bv1Rz411e7eJ的评论区才发现的。那么如果你的CUDA，cuDNN版本都对，只有Pytorch安装成了CPU的，那么先对它进行卸载，然后再安装
在这里插入图片描述

卸载方法：
找到刚才下载成功的三个东西，如果没在虚拟环境下下载Pytorch的话，他们在你的anaconda目录下的lib->site-packages，例如我的在D:\anaconda\Lib\site-packages，删除torch和torch-1.1.1+cpu.dist.info（torch-1.7.1+cu101.dist-info中的+ c u 101 表示 C U D A 10.1 ， +