win10 下重装anaconda并配置torch环境(cuda11.2)

本次重装是因为这几天安装torch, torchvision(安装torch+cuda时没有一起装上,手动安装后又连累了torch版本,再手动安装torch+cuda), almbumentations后出现以下报错
win10 下重装anaconda并配置torch环境(cuda11.2)_第1张图片
使用以下安装思路:

  1. 找到自己机器cuda版本适合的python版本
  2. 安装对应版本的anaconda(而不是最新版)
  3. 复制base环境新建一个conda环境
  4. 在新建的环境下pip安装torch
  5. 亲测有效

卸载anaconda

来到C盘下anaconda3的文件夹,运行Uninstall-Anaconda3.exe
win10 下重装anaconda并配置torch环境(cuda11.2)_第2张图片
漫长的等待后卸载完成。卸的非常干净,什么都没了。C盘立即多出20G。

确认自己要安装的Python版本和torch版本

anaconda中创建的不同环境对应了不同的Python版本。
安装anaconda自带的base环境里也自带了一个python版本。我们不能直接下载最新的anaconda,因为最新的anaconda的base版本中的pyhton太新了,会导致我们torch安装不上。
你可能会说自己创建一个环境不就好了?自己创一个空空的环境,一切都要自己安装; 如果可以直接复制base环境(有很多常用包已经安装),就省去了很多繁琐的装包过程。
这就是为什么要下载老版本的anaconda了。

Python版本的选择是根据这篇博客选择的。
win10 下重装anaconda并配置torch环境(cuda11.2)_第3张图片
先看自己的cuda版本,找到对应的Python版本。cuda不一定能一一对应,有两个原则

  • 找比自己低的版本
  • 自己是11.x的cuda,就不要找10.x的cuda版本。
  • 比如说我是cuda 11.2, 找cuda11.1是合适的。
  • 自己的cuda版本查看方式请自行谷歌(百度不太好用),网上很多。

python3.7 对应anaconda3-2020.02
一般下Python3.7就可以了。其他的对应版本也可以自行谷歌。

下载安装anaconda

https://repo.anaconda.com/archive/
找到和自己匹配的版本,机器类型,安装即可。

环境变量配置

换成自己anaconda路径

E:\Anaconda(Python需要) 
E:\Anaconda\Scripts(conda自带脚本) E:\Anaconda\Library\mingw-w64\bin(使用C with python的时候)
E:\Anaconda\Library\usr\bin 
E:\Anaconda\Library\bin(jupyter notebook动态库)

验证安装成功

终端输入conda --version, 正确显示版本。

更换anaconda默认保存位置

C盘资源太珍贵了┭┮﹏┭┮,这一步是为了把环境保存路径默认到D盘。
不care C盘容量的可以跳过了。。
参考链接

换源

查看现在的源,删除除了default之外的所有源,换上中科大源。依次使用以下命令:

conda config --show channels  # 查看
conda config --remove channels   # 删除上一步看到的除default之外的源
# 以下两条命令换源
conda config --add channels https://mirrors.ustc.edu.cn/anaconda/pkgs/free/
conda config --set show_channel_urls yes

创建虚拟环境

conda create -n dl --clone base

安装cuda

用上面找到的对应的torch版本,比如说是1.8.0,来到https://pytorch.org/get-started/previous-versions/, ctrl+F, 搜索1.8.0:

于是关闭梯子,尝试这条命令:

pip install torch==1.8.0+cu111 torchvision==0.9.0+cu111 torchaudio==0.8.0 -f https://download.pytorch.org/whl/torch_stable.html

等待一会,安装成功。

在pycharm里使用conda环境

files-settings-python interpreter
win10 下重装anaconda并配置torch环境(cuda11.2)_第4张图片
选择本地解释器
选择system interpreter
选择已有解释器(existed)
路径是conda安装路径下的envs/环境名/python.exe

接下来就可以跑深度学习咯~
欢迎大佬留言指正!

你可能感兴趣的:(python,pytorch,开发语言)