我最近在做毕业设计,任务是去复刻某个深度学习,然后再改进一些。本来我以为会是个挺轻松的任务,没想到光安环境就能折腾我这么久。
我用的是谷歌的 colab ,好像不管怎么弄,cuda 的环境都改不动,就算我卸载了,也还是 11.8 。
就很奇怪,要被 cuda 折磨疯了。
下面给出原代码给出的环境,各位觉得这个配置存在问题吗?或者有更简单的安装环境的方法吗?
PyTorch version: 1.8.0
CUDA version: 11.1
cudnn version: 8005
GPU: Tesla V100-SXM2-32GB
requirements.txt
matplotlib==3.4.2
pynvml==11.0.0
torchvision==0.8.0
pandas==1.3.2
numpy==1.20.3
imbalanced_learn==0.8.0
tqdm==4.62.1
torch==1.7.0
fairseq==0.10.2
transformers==4.10.2
imblearn==0.0
librosa==0.8.1
Pillow==8.3.2
pysndfx==0.3.6
scikit_learn==1.0