1.先下载然后手动安装
GPU版本安装Pytorch教程最新方法_pytorch gpu-优快云博客
就是如图: 这个博主是按照目录下载下来然后进行安装的,这样有好处就是可以避免安装错误(这里我没有选择直接命令安装,因为安装失败的概率较高。我选择的是下载安装包,再在anaconda里下载whl文件。)坏处就是稍微有点麻烦
2.直接命令行安装
安装CUDA与CUDNN与Pytorch(最新超级详细图文版本2023年8月最新)_cudnn12.2-优快云博客
这个博主就是直接用命令行安装的,感觉也很慢。
装好了过后测试报错为:
Traceback (most recent call last):
File "E:\yolov5-5.0\adgayujd.py", line 1, in <module>
import torch
File "D:\SoftwareData\Anaconda3\envs\yolov5_wzh\lib\site-packages\torch__init.py", line 29, in <module>_
from .torch_version import version as version
File "D:\SoftwareData\Anaconda3\envs\yolov5_wzh\lib\site-packages\torch\torch_version.py", line 3, in <module>
from pkg_resources import packaging # type: ignore[attr-defined]
ImportError: cannot import name 'packaging' from 'pkg_resources' (D:\SoftwareData\Anaconda3\envs\yolov5_wzh\lib\site-packages\pkg_resources__init__.py)
更改pack解决问题