0
点赞
收藏
分享

微信扫一扫

Pytorch(0)-环境搭建

Resin_Wu 2022-02-03 阅读 98

1.anaconda下载
2.查看是否有独立显卡以确定 Pytorch的下载参数
3.安装Pytorch

 


1.anaconda下载
 
Anaconda | Individual Edition


 
2.查看是否有独立显卡以确定 Pytorch的下载参数
 
3种查看方式 :
  1.任务管理器-GPU型号

  

  2.设备管理器

3.命令行
 

nvidia-smi


 显卡驱动网站:Official Drivers | NVIDIA
3.安装Pytorch
   网站选型:
 

#1.先用 anaconda prompt 建一个conda environment: 
conda create -n pytorch pyton=3.7  #创建
conda active pytorch               #激活环境
pip list                           #查看安装包


#2.在这个环境中安装 Pytorch 安装参数根据前面查看的硬件配置分(有显卡,没显卡)两种安装包
https://pytorch.org/
conda install pytorch torchvision torchaudio cudatoolkit=10.2 -c pytorch


  安装的时候看网速,我这花了十几分钟。

安装验证:
 

python         #进入python
import torch   #进入torch
torch.cuda.is_available()   #是否可以访问gpu



搞定!
 

 

 

 

 

 

举报

相关推荐

0 条评论