1.首先,确定自己显卡的算力---确定自己的显卡型号
点击任务管理器---性能
2.CUDA官网----https://en.wikipedia.org/wiki/CUDA----确定显卡算力(8.6)
3.确定CUDA Runtime (对应下面都支持)
4. 看自己的驱动CUDA Driver Version
搜索cmd(命令提示符)----nvidia-smi
5.最终确定适用CUDA 11.1-11.6
*******************************************************************************************
一、
1.安装显卡驱动最新版本--nvidia官网
2.打开命令窗口,输入nvidia-smi确定cuda driver版本(本人:11.6)
3.打开Pytorch官网,确定CUDA Runtime 版本
Previous PyTorch Versions | PyTorch,找到低于11.6版本的pytorch
二、
在创建的虚拟环境中安装Pytorch(安装Pytorch, 安装3个包,直接从官网找到对应的conda版本/pip版本文件)
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
三种途径:①直接anaconda prompt
②从pycharm终端(Terminal)直接输入
③本地下载,然后利用pip install 路径地址安装
(将地址复制到迅雷里,这里的路径地址是指下载后文件所在地址)
三、验证pytorch是否成功
①激活对应的虚拟环境(conda activate 虚拟环境名)
②输入conda list,看有没有pytorch或torch
③输入python
④输入import torch-----回车 输入torch.cuda.is_available()-----如果返回True,说明成功(可以使用GPU版本)
********每天进步一丢丢,知识来源:b站:我是小土堆(仅供学习使用,用于自己对知识的总结)