【已解决】懒人安装pytorch的GPU版本,三步解决

博主分享了在PyCharm中成功安装GPU版本PyTorch的过程,强调新建环境的重要性。详细记录了requirements.txt文件内容,包括所需库的版本,并提供了安装命令。遇到的问题主要是torch包版本与CUDA版本匹配,提醒读者根据错误提示调整。
摘要由CSDN通过智能技术生成

GPU版本的pytorch的安装折磨了我许久,终于在pycharm下运行成功,下面给大家提供我的包。一定新建环境,一定新建环境,一定新建环境。
我下载的cuda版本为11.1,如果版本不一样的同学可以跳过这篇文章了。
我的电脑最高支持11.2,,python版本为3.7,关于anaconda和cuda的下载,不再赘述,下面就是懒人版安装过程:


1.首先在E盘创建requirements.txt,内容如下:

absl-py==1.0.0
cachetools==5.0.0
certifi==2021.10.8
charset-normalizer==2.0.12
colorama==0.4.4
cycler==0.11.0
fonttools==4.32.0
google-auth==2.6.5
google-auth-oauthlib==0.4.6
grpcio==1.44.0
idna==3.3
importlib-metadata==4.11.3
kiwisolver==1.4.2
Markdown==3.3.6
matplotlib==3.5.1
numpy==1.21.6
oauthlib==3.2.0
opencv-python==4.5.5.64
packaging==21.3
pandas==1.3.5
Pillow==9.1.0
protobuf==3.20.0
pyasn1==0.4.8
pyasn1-modules==0.2.8
pyparsing==3.0.8
python-dateutil==2.8.2
pytz==2022.1
PyYAML==6.0
requests==2.27.1
requests-oauthlib==1.3.1
rsa==4.8
scipy==1.7.3
seaborn==0.11.2
six==1.16.0
tensorboard==2.8.0
tensorboard-data-server==0.6.1
tensorboard-plugin-wit==1.8.1
torch==1.9.0+cu111
torchaudio==0.9.0
torchvision==0.10.0+cu111
tqdm==4.64.0
typing_extensions==4.1.1
urllib3==1.26.9
Werkzeug==2.1.1
wincertstore==0.2
zipp==3.8.0

2.然后打开anaconda prompt,新建环境,进入新建的环境,然后下载一个git小工具,然后针对thop包进行专门下载,出错的话多试几次

conda create -n pytorch python=3.7
activate pytorch
conda install git
pip install --upgrade git+https://github.com/Lyken17/pytorch-OpCounter.git

3.安装requirements.txt里面的包,耐心等待即可

pip install -r e:\requirements.txt

如果报错,请根据报错提示操作,大概率是torch包的版本问题,大家去官网找一下CUDA11.1对应版本的命令,单独执行一下,如果报错,大概率还是版本问题,根据提示信息下载适合的版本即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值