GPU使用

首先环境安装(anaconda3+pytorch)

1:下载anaconda3

2:bash安装anaconda3

   这里会遇到conda不能用的问题,解决办法如下:zsh:command not found: conda

   解决方法:

   一: vim ~/.zshrc

   二:export PATH = /root/userfolder/anaconda3/bin:$PATH

  GPU使用_第1张图片

  三:source ~/.zshrc

3:安装完之后就需要安装pytorch环境啦

在此之前,我们可以更换我们的镜像源,比如更换清华、阿里等

安装pytorch的时候我最开始使用conda进行安装,可是出现的问题是一直在solving environment,所以我放弃了比较喜欢用的condo,开始使用pip进行安装,安装代码如下:

GPU使用_第2张图片

  

pip install torch==1.5.1+cu101 torchvision==0.6.1+cu101 -f https://download.pytorch.org/whl/torch_stable.html

最好祝大家学习进步。

 

你可能感兴趣的:(机器学习,python)