1.conda创建新环境:(一般python都是3.6,3.7,pycharm好像不支持3.5了)

conda create -n xxxxx python=3.7

2. 进入环境:

conda activate xxxx

3.安装cuda:(这里面叫cudacoolkit)

conda install cudatoolkit=10.2 -c https://mirrors.tuna.tsinghua.edu.cn/anaconda/pkgs/free/win-64/

后面放的是清华镜像源,可以自己打开链接看一下

4.安装对应版本的cudnn:

这次搞了很久,好像是悟了,原来不是每个版本的cudnn都能装进conda环境

main/win-64 (anaconda.com)icon-default.png?t=M4ADhttps://repo.anaconda.com/pkgs/main/win-64/ctrl+F,我们可以看见就只有部分cuda和cudnn(之前一直想下cuda9.2和cudnn7.2.1,一直没下下来还以为是网太差,这个库里都没有,伤心)

 

Logo

欢迎来到由智源人工智能研究院发起的Triton中文社区,这里是一个汇聚了AI开发者、数据科学家、机器学习爱好者以及业界专家的活力平台。我们致力于成为业内领先的Triton技术交流与应用分享的殿堂,为推动人工智能技术的普及与深化应用贡献力量。

更多推荐