Jupyter Notebook添加新的torch环境经验
已有的软件基础 pycharm ,vscode ,anaconda, Jupyter Notebook,原来的环境不知道为什么在跑反向传播loss.backward()时候崩溃了,所以就想配个新环境,主要总结一下踩过的坑
我比较喜欢用pycharm的terminal装包,喜欢用vscode运行jupyter程序,所以没有这两个软件也没关系,有anaconda就可以了
-
在pycharm右下角add interpreter中新建环境,我选择的是anaconda的python3.6环境
也可以用命令行建立新环境,我设的环境名为5tsp
conda create -n 5tsp python=3.6
-
在环境中安装pytorch
本人cuda版本为11.2,所以按道理只要比11.2低的版本都可以安装
发现torch新版本如1.9.0 1.8.0这类的版本安装包过大
容易安装不上,或者安装结束时死机,亲测用pycharm的命令行比anaconda自带命令行成功率高一些,不过也会卡死,而且开始时候1.9.0好不容易装上了,但是遇到了个报错
name '_C' is not defined
找了一圈办法没解决,所以最后选择装了低版本,torch1.7.1
conda install pytorch==1.7.1 torchvision==0.8.2 torchaudio==0.7.2 cudatoolkit=11.0 -c pytorch
如果安装高版本,需要防止安装过慢而超时
pip --default