因为在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.
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
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
安装成功!!
文章出处登录后可见!