Anaconda3+Pytorch+Pycharm配置过程

1.师弟已经把Anaconda装好了,python版本是python3.6,CUDA版本是10

2.卸载原来的pytorch和torchvision

   win+R,输入cmd,pip uninstall torch,结束后再输入pip uninstall torchvision

3.参考了https://blog.csdn.net/qq_33183517/article/details/90484840,打开Anaconda Prompt:

  (1)创建Pytorch环境,conda create -n pytorch python=3.6

  (2)激活Pytorch环境,activate pytorch (退出:deactivate pytorch)

  (3)按照下图来安装pytorch,注意pip3改为pipAnaconda3+Pytorch+Pycharm配置过程_第1张图片

   (4)注意:安装torch1.0.1后,对应的torchvision版本一定要是0.2.1,不能装0.3,否则import torchvision会报错:pip install torchvision==0.2.1。有时安装pytorch会莫名中断,重新开始或升级pip即可,安装成功后,输入python,输入import torch,输入import torchvision,三次没有报错,就说明安装成功了。

4.随后参考了https://blog.csdn.net/ruotianxia/article/details/100066181

   (1)新建一个Pycharm工程,然后File>>>>Settings>>>>Project Interpreter, 这时会发现项目解释器是Pycharm自带的或者是Anaconda自带的python.exe。然而,我们需要使用pytorch的库,所以点击右边代表‘设置’符号,点击ADD, 选中Existing environment,找到目录并选中:C:\Users\Administrator\Anaconda3\envs\pytorch\python.exe作为该项目的解释器。

   (2)验证Pytorch是否导入Pycharm环节,输入:

            from __future__ import print_function

            import torch

            x = torch.rand(5, 3)

            print(x)

           点击运行没报错就说明可以了

  (3)验证Pytorch是否可以使用GPU和CUDA

           import torch

           torch.cuda.is_available()

           点击运行没报错就说明可以了

        

           

 

你可能感兴趣的:(深度学习,计算机视觉)