如何正确下载PyTorch、CUDA版本!!

因为在torch框架里经常出现NVIDIA、CUDA、PyTorch这几个不兼容,等等一些让人头疼的问题。这里总结正确下载pytorch的方法。

目录


安装pytorch

一.查看自己cuda版本

方法一:

 方法二:

CMD中输入:

nvidia-smi

 

我的是CUDA版本12.0,版本向下兼容,意思就是CUDA 12.0及以下版本的都可以安装

二.打开pytorch的官网:PyTorchAn open source machine learning framework that accelerates the path from research prototyping to production deployment.如何正确下载PyTorch、CUDA版本!!https://pytorch.org/(官网网址:https://pytorch.org/)

 复制代码:

pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu117

安装CUDA

官网:CUDA Toolkit 11.7 Downloads | NVIDIA DeveloperResources CUDA Documentation/Release NotesMacOS Tools Training Sample Code Forums Archive of Previous CUDA Releases FAQ Open Source PackagesSubmit a BugTarball and Zip Archive Deliverables如何正确下载PyTorch、CUDA版本!!https://developer.nvidia.com/cuda-11-7-0-download-archive?target_os=Windows&target_arch=x86_64&target_version=10&target_type=exe_local

 选择 11.7的对应的进入安装,有点大2.5g,耐心等待呀 

 打开exe,自定义

全选

记住安装位置,默认位置

配置环境变量

在系统变量里新建这四个,后边是v11.7安装的位置

CUDA_PATH
CUDA_PATH_V11_6
NVCUDASAMPLES_ROOT
NVCUDASAMPLES11_0_ROOT

 验证使用CMD查看CUDA是否安装成功命令:nvcc –version或nvcc -V

nvcc --version
nvcc -V

安装成功!! 

文章出处登录后可见!

已经登录?立即刷新

共计人评分,平均

到目前为止还没有投票!成为第一位评论此文章。

(0)
扎眼的阳光的头像扎眼的阳光普通用户
上一篇 2023年7月13日
下一篇 2023年7月13日

相关推荐