Anaconda(python3.9.7)+pytorch1.10+Cuda11.3

被环境折磨了一个早上,多次尝试后终于成功,记录一下怎么装。结论是,不要贪新!并不是cuda新的好!

 

1.安装anaconda,网上很多教程就不说了。主要是后面

2.准备:cuda每个版本地址,下载11.3

CUDA Toolkit Archive | NVIDIA DeveloperPrevious releases of the CUDA Toolkit, GPU Computing SDK, documentation and developer drivers can be found using the links below. Please select the release you want from the list below, and be sure to check www.nvidia.com/drivers for more recent production drivers appropriate for your hardware configuration.https://developer.nvidia.com/cuda-toolkit-archive3.更新显卡驱动,安装界面不需要装显卡驱动里面的Experience.这是一个管理游戏的界面用不着。

4.安装下载好的cuda。选择自定义,第一个cuda必须勾选,后面的跟你的显卡驱动对比,一般你的驱动是最新的话就去掉后面的两个勾勾。

5.重要说明!!!!

此处是最高支持cuda版本,这里是11.5.107,然而直接去cuda官网下载最新的可能会比这个要高,就是这里!!!不要直接在这里下载最为保险!CUDA Toolkit 11.5 Update 1 Downloads | NVIDIA Developerhttps://developer.nvidia.com/cuda-downloads

Anaconda(python3.9.7)+pytorch1.10+Cuda11.3_第1张图片

评估算力网页

CUDA GPUs | NVIDIA Developericon-default.png?t=LA92https://developer.nvidia.com/cuda-gpus#collapseOne

pytorch官网

Start Locally | PyTorch An open source machine learning framework that accelerates the path from research prototyping to production deployment.https://pytorch.org/get-started/locally/#windows-prerequisites-2Anaconda(python3.9.7)+pytorch1.10+Cuda11.3_第2张图片

后面的-c  python去掉,变成

conda install pytorch torchvision torchaudio cudatoolkit=11.3

 激活自己的虚拟环境,输入指令,下载就行!

 Anaconda(python3.9.7)+pytorch1.10+Cuda11.3_第3张图片

cmd 输入

nvcc -V

可以看到cuda11.3驱动,正常装上最高支持cuda11.5.107的驱动的显卡上。

然后最后。

import torch
torch.cuda.is_available()

Anaconda(python3.9.7)+pytorch1.10+Cuda11.3_第4张图片

 True!!!!!!!!!!!!!!!!!!!!!!!!!!

你可能感兴趣的:(深度学习,pytorch)