GPU安装教程(付费内容,且看且珍惜)
- 首先conda info -e 获取当前的环境有哪些
- 然后选到你的环境 conda activate py37(环境名字),接着就是pip uninstall torch torchvision torchaudio如果你之前有包的话就要先删掉,没有的话执行以下命令也没有事
Ps:这个黄字红线的意思是 如果发现 torchaudio 没有安装,则跳过与之相关的功能或安装步骤。在这里就是没安装然后执行要卸载的命令就直接跳过了
- 接着就是看当前cuda版本:nvidia -smi
- nvcc -V 是一个用来检查 NVIDIA CUDA的版本的命令。命令行 nvcc -V 检查CUDA,python-V 查看当前环境python的版本
-
- 直接命令行下载,但是非常慢,之前下过这样的,我挺蠢的,没错。
- 然后重点来了,上download.pytorch.org/whl/torch_stable.html找匹配的,很好,你问我怎么找匹配的,我也有点稀里糊涂,等我学习一下,下期出。所以用网站下载download.pytorch.org/whl/torch_stable.html,找好自己电脑适合的,可以直接下,也可以用迅雷下
- 之后就是到当前环境终端pip install C:\Users\samuel\Downloads\torch-1.13.1+cu117-cp37-cp37m-win_amd64.whl 把下载下来的轮子给下到当前环境里去。然后在执行之前一长串的下载很慢的那个命令。
为大家送上整体图: