conda配置安装pytorch

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

猜你喜欢

转载自www.cnblogs.com/jaysonteng/p/12544405.html