因为在torch框架里经常出现NVIDIA、CUDA、PyTorch这几个不兼容,等等一些让人头疼的问题。这里总结正确下载pytorch的方法。
目录
安装pytorch
安装CUDA
安装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.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 Deliverableshttps://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
文章来源:https://www.toymoban.com/news/detail-557792.html
安装成功!! 文章来源地址https://www.toymoban.com/news/detail-557792.html
到了这里,关于如何正确下载PyTorch、CUDA版本!!的文章就介绍完了。如果您还想了解更多内容,请在右上角搜索TOY模板网以前的文章或继续浏览下面的相关文章,希望大家以后多多支持TOY模板网!