Mac下pycharm如何安装pytorch「建议收藏」

全栈程序员-站长 • 2022年8月28日下午2:46 • 未分类 • 阅读 9

通过虚拟环境virtualenv安装pytorch

首先提及，下面的命令，我都是直接在macos上的pycharm中的terminal执行的

1、安装virtualenv虚拟环境：

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple –upgrade virtualenv

或者 python -m pip install -i https://pypi.tuna.tsinghua.edu.cn/simple –upgrade virtualenv

2、创建虚拟环境

virtualenv –system-site-packages -p python3 targetDirectory

注意需要找到virtualenv 安装的位置，然后targetDirectory是你想要存放虚拟环境的路径，下面是我的使用，我自己在pycharm的项目路径下，新建了venv目录并在该目录下为我即将创建的tensorflow环境，建了子目录，以后就在这个venv下，管理我的所有虚拟环境

./virtualenv –system-site-packages -p python3 ~/PycharmProjects/venv/pytorch01/

3、激活虚拟环境

source ~/PycharmProjects/venv/pytorch01/bin/activate

显示如下图

Mac下pycharm如何安装pytorch「建议收藏」

4、在激活的虚拟环境中安装 Pytorch 及其所有依赖

pip install torch torchvision
鉴于上面的命令实在太慢了，耗时长，我用了清华的镜像，如下

pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch torchvision

这样发现可能报错，可以换个豆瓣的源试试

pip install torch torchvision -i http://pypi.douban.com/simple/ –trusted-host pypi.douban.com

其中–trusted-host pypi.douban.com 的作用是获得ssl证书的认证，这个可以先不加，不行的话再加上

同理，清华的也可以加上–trusted-host 试一试

然后版本记得匹配就好，我用的是torch==1.6.0 torchvision==0.7.0

5、验证安装效果

在命令行输入python，进入python环境，import pytorch 如果没有报错，就安装ok

下面的简单语句，可以测试一下输出

import torch

print(torch.rand(2,3))

恭喜各位开启崭新的pytorch之旅吧。如果安装有问题，欢迎随时交流哦。

发布者：全栈程序员-站长，转载请注明出处：https://javaforall.net/175082.html原文链接：https://javaforall.net