突然发现办公室公机居然是1060!!!好开心啊哈哈哈哈哈好容易满足TAT
记录一下安装pytorch gpu版本的全过程,
首先电脑配置是:cpu:i7-8700K,gpu:gtx 1060 6BG
目录
安装anaconda3和pycharm
首先,按照这里安装pycharm最新版。
然后是anaconda3的最新版: 下载地址
使用anaconda安装cuda和cudnn
检查电脑里面的cuda版本
在电脑桌面右键NVIDIA控制面板,点击帮助->系统信息->组件
这里我的cuda版本是11.1
安装对应的cudnn
直接在anaconda3的终端输入conda install cudnn 安装对应的cudnn
使用anaconda安装pytorch-gpu
创建新环境
输入conda create --name pytorch python=3.7 创建一个python3.7名为pytorch的环境
激活新环境并在此环境下安装pytorch
输入:activate pytorch 激活刚刚创建的新环境
在pytorch官网link 复制命令:conda install pytorch torchvision torchaudio cudatoolkit=11.1 -c pytorch -c conda-forge 到终端安装pytorch
最后conda list 查看是否安装成功
done~