Pytorch环境的配置及安装

最新推荐文章于 2024-07-03 15:36:08 发布

雯雯Didi

最新推荐文章于 2024-07-03 15:36:08 发布

阅读量246

点赞数

分类专栏： Pytorch 文章标签： pytorch 深度学习

本文链接：https://blog.csdn.net/weixin_44876084/article/details/120234439

版权

1 篇文章 0 订阅

订阅专栏

遇到不同的项目，需要使用不同版本的环境，比如该项目要用到Pytorch 0.4，另一个项目要用到Pytorch 1.0，如果卸载了0.4版本，安装了1.0版本，那下一次，再碰到0.4版本，就需要卸载1.0版本，安装0.4版本。

Anaconda集成的 conda 包就能解决这个问题，即造出两个屋子，相互隔离。一个屋子放 0.4 版本，一个屋子放 1.0 版本。需要哪个版本，就进哪个屋子工作。

首先使用 conda 指令创建一个屋子，叫做Pytorch

指令如下：

conda create -n pytorch python=3.6

done
#
# To activate this environment, use
#
#     $ conda activate pytorch
#
# To deactivate an active environment, use
#
#     $ conda deactivate

检验是否下载好

pip list

python
import torch   #未报错即可
torch.cuda.is_available()   #检查pytorch是否可以使用GPU，返回True即可

关注