前言:我是一个新手,如果我下面有说的不对,请大家多多指教。我只有一张3070显卡,最近在入门机器学习,然后网上配置机器学习环境的基本都不是用30系显卡的,所以大部分的教学都是配置CUDA10.2以下的版本的,而30系显卡似乎又不支持CUDA10.2以下,所以我也是摸着石头过河,一连配置了很长时间。(不是我不愿意用旧版本,我也想偷懒,照着搬环境,只是这用的是一张3070,就很让人难办..)
先配置了一个python3.7的环境,这个基本没什么难度。我配了一个叫TF的环境。
期间穿插了配置TensorFlow+Cuda+Cudnn,虽然也有磕磕碰碰,但并不会说很不顺利,唯独在配置pytorch的时候,是直接在pytorch的官网取得CUDA11.1:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1 -c pytorch -c conda-forge
然后直接在我配置的TF环境下运行,然后总是在下载 pytorch 和 cudatoolkit 时候中断,我试过很多办法,比如课雪商罔,修改失败重试的时间等,都不顶用,而且这个每次等到他中断了,才知道是失败了,这个时间成本实在是太高了。
我在网上看了各种办法,其实我这个也是老掉牙的办法,就是直接导入了清华源
正文
一、打开 cmd 进入TF
activate TF
二、直接运行以下代码(这是清华源的网址)
conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch/win-64/
这个是在旧的办法上改良而来的,以前都是 conda config --add channels https://mirrors.tuna.tsinghua.edu.cn/anaconda/cloud/pytorch,我点进去研究了一下,并猜测是后来更新了,细分了一下各个版本,并且我加多了一个分组,就有了上面这行。
三、设置一下 超时时间
pip --default-timeout=1000 install requests
这一行其实我一直在想要不要添加,因为其实一开始我试了一下并没有什么效果,解决不了 下载中断和失败的问题,但保险起见还是加上吧,不然每次等到他失败,这个时间成本实在是太高了。
四、把官网复制来的CUDA11.1的下载代码后面的 -c pytorch -c conda-forge 删掉,然后复制进去运行,即运行:
conda install pytorch==1.8.0 torchvision==0.9.0 torchaudio==0.8.0 cudatoolkit=11.1
如果只剩 pytorch 和 cudatoolkit 没装上,就直接:
conda install pytorch==1.8.0 cudatoolkit=11.1
接下来 等 就成功了。
检验成果:
一、进入python
二、输入环境与检验
import torch
torch.cuda.is_available()
希望大家都能得到这个结果。
关于Anaconda+Cuda+Cudnn+TensorFlow+Pytorch+Pycharm的环境配置,我实在是在网上受教太多了,希望如果有下载失败的人也能在我这里得到帮助。