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改为pip

   (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()

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

        

           

 

评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值