本文主要是介绍在Anaconda3虚拟环境中安装cuda、cudnn和pytorch,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!
1.打开Anaconda Prompt进入虚拟环境,然后依次添加中科大镜像源
参考:https://blog.csdn.net/weixin_48424686/article/details/116712394
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/main/
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/cloud/pytorch/
2.下载cuda与cudnn(网速不错的情况下可以直接镜像 conda install...,网不好的情况下可以离线下载再安装,此处为第二种情况)
根据自己的系统,在 中科大镜像 中找到所需要的安装包,下载cudatoolkit10.2和cudnn7.6.5
我的是64位系统,进入 win-64/
将安装包复制到 D:\ProgramData\Anaconda3\pkgs(根据自己实际的安装路径而定,一般安装包都在pkgs文件夹下)
3.打开Anaconda Prompt,安装这两个包
conda install D:/ProgramData/Anaconda3/pkgs/cudatoolkit-10.2.89-h74a9793_1.tar.bz2
conda install D:/ProgramData/Anaconda3/pkgs/cudnn-7.6.5-cuda10.2_0.tar.bz2
安装后输入 conda list,显示cuda与cudnn安装成功
4.下载pytorch(镜像)
在pytorch官网上寻找CUDA10.2所对应的指令
输入第四条指令(一定要删除 -c pytorch),等待几分钟下载完成
conda install pytorch==1.5.0 torchvision==0.6.0 cudatoolkit=10.2
5.最后输入以下三条指令,若返回TRUE,则安装成功
python
import torch
torch.cuda.is_available()
这篇关于在Anaconda3虚拟环境中安装cuda、cudnn和pytorch的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!