前言:我是一个新手,如果我下面有说的不对,请大家多多指教。我只有一张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 时候中断,我试过很多办法,