Pytorch安装篇（自用）

1）查看自己电脑的cuda版本

使用win+R，输入cmd，打开命令提示符：

nvcc -V
nvcc --version

 

可以看到我的cuda版本时11.8

2）查看可用的cuda版本对应可用的pytorch

访问pytorch官网，查看与相应的cuda适配的版本。

 

我的cuda版本11.8对应可用的pytorch版本为2.0.0/2.0.1/2.1.0.......等

3）查看对应的python版本

官网：pytorch的github官方文档：https://github.com/pytorch/vision#installation

 

我准备下载的pytorch版本是2.0.0对应python环境3.8<=python<=3.11

4）环境管理

创建环境

conda create -n 环境名 python=3.8

 

进入该环境

conda activate 环境名

 

5）安装pytorch

寻找到对应的pytorch环境

 官网：Previous PyTorch Versions | PyTorch

pip install torch==2.0.0 torchvision==0.15.1 torchaudio==2.0.1 --index-url https://download.pytorch.org/whl/cu118

6）检查pytorch下载结果

检查pytoch是否安装成功，可以和我们的cuda相匹配吗

import torch     # 如果pytorch安装成功即可导入
print(torch.cuda.is_available())     # 查看CUDA是否可用
print(torch.cuda.device_count())     # 查看可用的CUDA数量
print(torch.version.cuda)     # 查看CUDA的版本号
exit()    #退出

 

（持续更新中......）