武丢丢

细想全是问题,去做才有答案

0%

GPT2环境配置

环境配置记录

本地配置方法:去pytorch官网找到要下载的历代版本,比如本次配置版本为

conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

直接使用这个命令即可下载pytoch相关包以及cuda包,此处cuda包的版本不需要非得和电脑上安装的cuda软件版本一致。最后在下载一个cudnn的包即可

速度太慢

直接使用上述命令的速度太慢,故采用本地下载方式。

1.去清华镜像网站找到对应python版本的pytorch,torchvision和torchaudio包的文件并下载,并使用下述命令下载到虚拟环境:

conda install --use-local C:\Users\武丢丢\Downloads\文件名

2.使用conda search cuda –info命令,得到适配的cuda版本及下载网址,cudnn同理。得到cuda和cudnn的本地包之后,用步骤1命令下载到虚拟环境。

3.下载完进入虚拟环境后运行import torch会报错shm.dll找不到,这是在虚拟环境中再次运行pytorch官方命令:(这次会快很多

conda install pytorch==2.2.0 torchvision==0.17.0 torchaudio==2.2.0 pytorch-cuda=11.8 -c pytorch -c nvidia

即可解决错误

注意

若遇到conda无法创建新环境的问题,重启电脑多半可以解决