深度学习环境搭建中python、torch、torchvision、torchaudio、cuda的安装_torch-torchvision-python安装

作者：我家小花儿 | 2024-02-22 05:08:28

踩

torch-torchvision-python安装

深度学习遇到的第一个阻碍——环境搭建

跑学长给的框架遇到了pycharm报错
在这里插入图片描述
查阅网上相关信息后，这是因为要用GPU跑代码，没连接成功。

尝试了各种办法，历经4天终于解决，以此文来记录。

安装

需要安装python、torch、torchvision、torchaudio、cuda
注意这5个的版本都需要对应，在这里查看版本对应关系：https://pytorch.org/get-started/previous-versions/
当然也可以在网上查找对应关系
在这里手动安装、torch、torchvision、torchaudio：https://download.pytorch.org/whl
但是我碰到一个坑就是按照cuda10.2对应的版本安装好之后说驱动版本太低，于是我有换成cuda11.3，然后成功了。
最终我才用的配置为:

python	3.8
torch	1.10.1
torchaudio	0.10.1
torchvision	0.11.2
cuda	11.3

验证

在anaconda中输入

python
import torch
torch.cuda.is_available()

1
2
3
4

终于得到了心心念念的TRUE
在这里插入图片描述

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/我家小花儿/article/detail/128631?site