1、新建环境,避免与基础环境出现冲突
conda create -n pytorch python=3.6
2、切换到pytorch环境
conda activate pytorch
3、安装基本框架
conda install pandas
......
4、安装pytorch
1)进入官网https://pytorch.org/get-started/locally/
2)根据自己系统 和cuda版本,选好对应选项
3)复制Run this Command中的代码到pytorch环境终端,开始安装。
4)cuda版本查看:
a).windows:
桌面右键-->NVDIA控制面板-->帮助-->系统信息
组件-->NVCUDA.DLL查看CUDA版本
b)linux:
打开终端,输入:nvcc -V
或者:
CUDA:
cat /usr/local/cuda/version.txt
cudnn:
cat /usr/local/cuda/include/cudnn.h | grep CUDNN_MAJOR -A 2