Pytorch环境配置2.0.1+ Cuda11.7

查找cuda、cudnn、Pytorch(GPU)及cuda和NVIDIA显卡驱动对应关系

查询可支持的最高cuda版本

bash 复制代码
nvidia-smi

查看支持的cuda的版本 CUDA版本对应表

我的显卡驱动是Driver Version:535.40.,那么左边对应的CUDA都可以兼容

右上角为CUDA 版本,可以看到该电脑可以支持的cuda版本最高是12.3,驱动是向下兼容的,所以cuda版本小于等于12.3的都可以安装上(建议11.8版本可以兼容大部分的库)

查看自己是否已经安装CUDA

bash 复制代码
nvcc -V

显示如下,表明已经安装

如果未安装,安装Cuda驱动工具

Cuda硬件驱动工具包安装教程
pytorch gpu版本的最全安装教程,含环境配置、CUDA(百度云下载)安装程序失败解决方案

如果想配置不同版本的cuda,可以看这个
如何在电脑上安装多个版本的CUDA

创建pytorch的环境

bash 复制代码
conda create -n pytorch_GPU python=3.9
conda activate pytorch_GPU

安装torch2.0.0+cuda11.7

官网 找到了pytorch1.7版本的conda安装命令

bash 复制代码
conda install pytorch==2.0.1 torchvision==0.15.2 torchaudio==2.0.2 pytorch-cuda=11.7 -c pytorch -c nvidia

查询torch版本情况

python 复制代码
import torch # 如果pytorch安装成功即可导入
print(torch.cuda.is_available()) # 查看CUDA是否可用
print(torch.cuda.device_count()) # 查看可用的CUDA数量
print(torch.version.cuda) # 查看CUDA的版本号
print(torch.__version__)  #注意是双下划

参考文献

https://blog.csdn.net/weixin_42184315/article/details/134259075

相关推荐
taxunjishu5 分钟前
DeviceNet 转 Modbus TCP 协议转换在 S7-1200 PLC化工反应釜中的应用
运维·人工智能·物联网·自动化·区块链
kalvin_y_liu30 分钟前
智能体框架大PK!谷歌ADK VS 微软Semantic Kernel
人工智能·microsoft·谷歌·智能体
爱看科技33 分钟前
智能眼镜行业腾飞在即,苹果/微美全息锚定“AR+AI眼镜融合”之路抢滩市场!
人工智能·ar
猫头虎2 小时前
如何查看局域网内IP冲突问题?如何查看局域网IP环绕问题?arp -a命令如何使用?
网络·python·网络协议·tcp/ip·开源·pandas·pip
沿着路走到底2 小时前
python 基础
开发语言·python
Juchecar3 小时前
LLM模型与ML算法之间的关系
人工智能
FIN66683 小时前
昂瑞微:深耕射频“芯”赛道以硬核实力冲刺科创板大门
前端·人工智能·科技·前端框架·信息与通信·智能
benben0443 小时前
京东agent之joyagent解读
人工智能
LONGZETECH4 小时前
【龙泽科技】汽车动力与驱动系统综合分析技术1+X仿真教学软件(1.1.3 -初级)
人工智能·科技·汽车·汽车仿真教学软件·汽车教学软件
烛阴4 小时前
武装你的Python“工具箱”:盘点10个你必须熟练掌握的核心方法
前端·python