Windows Pytorch环境配置
一. 安装Anaconda
官网下载安装,具体步骤这里不详细展开
二. 打开Anaconda prompt
1. 新建名为Pytorch的环境,python=3.6,并激活
conda create -n Pytorch python=3.6
激活名为Pytorch的环境
conda activate Pytorch
查看显卡驱动的CUDA版本,以便对照安装相应的Pytorch
nvidia-smi
三. 安装Pytorch
复制官网的安装命令(继续在上面的命令行终端中运行conda xxxx -c pytorch)
如果换源了,需要删除掉安装命令中的’-c pytorch’,否则还是使用的官方源
四. 验证
五. Pycharm中配置Pytorch环境