一.安装适合的CUDA与CUDNN
安装教程 : 安装CUDA和CUDNN
二.安装 torch torchvision torchaudio
进入 PyTorch官方网站 网站,查看符合cuda版本的安装命令。
1.在线安装
如果CUDA的版本为11.2,就在官网查找带有 cu112 关键字符的安装命令,如果没有,可以安装 cu111 或更低的版本,比如如下命令行。
pip install torch==1.10.0+cu111 torchvision==0.11.0+cu111 torchaudio==0.10.0 -f https://download.pytorch.org/whl/torch_stable.html
2.本地安装(推荐)
由于某些原因,使用方法一安装时下载速度可能极慢,导致安装失败,此时选择本地安装就行。
(1).下载安装文件
进入 官方安装包 网站,直接 Ctrf+F ,输入 cu111 (或其他版本)关键字,在查找结果找到符合系统、符合python版本的两个whl文件:
- torch 的 cu111/torch-xxx.whl
- torchvision 的 cu111/torchvision-xxx.whl
直接右键点击这两个文件使用迅雷下载即可
(2).安装
将下载好的两个文件,复制到新建好的项目文件夹中,在命令行窗口,直接输入下面两个命令即可
pip install cu111/torch-xxx.whl
pip install cu111/torchvision-xxx.whl
三.检查是否成功安装
在项目下建立.py文件,输入一下命令,成功显示对应信息即可。
import torch
import torch.nn as nn
print(torch.cuda.is_available()) # True
print(torch.__version__) # 1.9.1+cu111
print(torch.cuda_version) # 11.1
print(torch.cuda.current_device()) # 0
print(torch.cuda.device_count()) # 1