windows10系统安装cuda版pytorch的一般步骤:
1、首先,查看本机是否有nvidia的GPU卡,并确定nvidia的GPU卡的型号。
2、其次,去nvidia官网下载对应显卡的最新版的驱动,下载地址为:https://www.nvidia.cn/drivers/lookup/。
下载驱动之后完成安装,之后在操作系统桌面右键单击-nvidia控制面板-系统信息-组件下面的NVCUDA64.dll一行可以看到该版本驱动支持的CUDA的最高版本,如我的电脑显示的是CUDA 11.4.557。
3、接着,去nvidia官网下载对应版本的CUDA Toolkit并行计算工具包,如适配本机显卡的CUDA Toolkit 11.4.4,下载地址为:https://developer.nvidia.com/cuda-toolkit-archive。
下载之后完成安装。并配置好CUDA的环境变量。配置方法参考链接:https://www.cnblogs.com/rainbow70626/p/6498714.html。
4、再次,去nvidia官网下载与CUDA Toolkit并行计算工具包对应的cuDNN库,Download cuDNN v8.2.4 (September 2nd, 2021), for CUDA 11.4,下载地址为:https://developer.nvidia.com/rdp/cudnn-archive。
下载之后完成安装。安装方法参考链接:https://blog.csdn.net/weixin_44779079/article/details/141528972。
5、下载cuda版pytorch离线安装包,下载地址为:https://download.pytorch.org/whl/torch/或https://download.pytorch.org/whl/torch_stable.html。比如我下载的是torch-1.12.1+cu113-cp39-cp39-win_amd64.whl,该文件的文件名中torch-1.12.1代表1.21.1版的torch,cu113代表cuda11.3版本,cp39-cp39代表python3.9版本,
6、下载Anaconda,下载地址为:Anaconda官网https://www.anaconda.com/download或清华大学镜像https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/,本机下载的是Anaconda3-2022.10-Windows-x86_64.exe。为啥下载该版本的Anaconda3?是因为它内嵌的python版本是3.9,与前一步下载的cuda版pytorch安装包项匹配。
这一过程可以在安装Anaconda3过程中,Anaconda安装包界面上看到python版本。下载Anaconda之后完成安装。
7、打开Anaconda prompt创建一个虚拟环境、激活虚拟环境、安装离线版的pytorch、验证gpu版pytorch是否安装成功。
#pytorch各个版本下载地址
pytorch的官方下载地址:https://download.pytorch.org/whl/
#anaconda各个版本下载地址
清华大学开源软件镜像站:https://mirrors.tuna.tsinghua.edu.cn/anaconda/archive/
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
一、离线安装
0、从链接https://download.pytorch.org/whl/torch/或https://download.pytorch.org/whl/torch_stable.html下载对应的CUDA版本的pytorch并保存到指定的文件夹。
1、打开Anaconda prompt创建一个虚拟环境:名为pytorch
conda create --name pytorch python=3.6 anaconda
2、激活该虚拟环境
activate pytorch
3、安装torch。找到之前下载的pytorch的绝对路径,E:\torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl,则直接用以下命令安装:
pip install E:\torch-1.7.0+cu101-cp36-cp36m-win_amd64.whl
4、上述安装过程中可能需要联网下载某些python包,直至提示pytorch安装成功。
5、判断gpu版pytorch是否安装成功。首先输入命令python切换到python运行环境,然后在输入以下程序,根据程序运行结果判断gpu版pytorch。
import torch
torch.cuda.is_available()
print("PyTorch version:",torch.__version__)
二、在线安装
去pytorch官网按命令行下载,不过比较慢,不推荐,传送门
1、根据自己的电脑(windows,Linux,mac选择)
2、使用conda安装
3、python编译环境
4、cuda版本号,这个根据自己之前安装的cuda版本号来确定
由于我的cuda:11.3,所以就选择这个,
下载:conda install pytorch torchvision torchaudio cudatoolkit=11.3 -c pytorch
原文链接:https://blog.csdn.net/HYilalala/article/details/121128944
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
# 安装时指定版本号
# CUDA 11.3
pip install torch==1.12.1+cu113 torchvision==0.13.1+cu113 torchaudio==0.12.1 --extra-index-url https://download.pytorch.org/whl/cu113
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
ERROR: Could not find a version that satisfies the requirement torch==xxx+cuxxx解决方案:
1、https://blog.csdn.net/weixin_46653651/article/details/140936271
2、https://blog.csdn.net/leva345/article/details/119836011