程序员

ML Note1 使用Conda安装PyTorch

2020-05-03  本文已影响0人  Ryker的3D龙

目的

记录实践过程,便于日后回顾,哪天环境重装了还得翻来看

步骤

  1. 安装pytorch到编译环境
conda env list
conda create -n pytorch python=3.7
activate pytorch
conda install pytorch torchvision cudatoolkit=10.1 -c pytorch

此代码可从 https://pytorch.org/get-started/locally/ 按需取得,下载速度可观

  1. pycharm中使用此conda环境解释器
上一篇下一篇

猜你喜欢

热点阅读