1、查看自己电脑支持的最高cuda版本
2、下载anaconda
3、通过anaconda prompt创建虚拟环境,小编python版本为3.9,cuda10.2。为什么要创建虚拟环境呢,是因为我在base环境安装好cuda和cudnn后,安装pytorch总是失败。但是在新建虚拟环境中可以成功安装。
4、安装对应版本的cuda以及下载cudnn。大家可以搜索到各个组件之间版本的对应关系。
5、最后在新建环境中下载pytorch,也要对应版本。小编在最后一步下载pytorch一直不成功,看了视频才解决掉。
!!!安装pytorch命令我从第二个,换为了第一个才最终成功下载pytorch。大家注意区分两种安装命令。找到不同,之后去找到跟第一个命令相同去安装pytorch。
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
conda install pytorch==1.10.0 torchvision==0.11.0 torchaudio==0.10.0 cudatoolkit=10.2 -c pytorch
!!!!
小编安装pytorch语句:
pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html
给大家推荐一个教程
Python深度学习:安装Anaconda、PyTorch(GPU版)库与PyCharm_哔哩哔哩_bilibili
大家没必要跟他去自己找torch,torchvision torchaudio的包,先试一下pip install torch==1.10.1+cu102 torchvision==0.11.2+cu102 torchaudio==0.10.1 -f https://download.pytorch.org/whl/cu102/torch_stable.html这种类型的语句可不可以,不可以的话可以学一下,小编是直接使用下载命令下载成功了。