17427 1 year ago
parent 29b60302b0
commit 209dbe8474

@ -0,0 +1,16 @@
###### 建议pip安装
# 首先卸载
pip3 uninstall torch -y
pip3 uninstall torchvision -y
pip3 uninstall torchaudio -y
# 然后安装
pip3 install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 -f https://download.pytorch.org/whl/cu121/torch_stable.html
###### conda安装
# 首先卸载完全卸载pytorch
conda uninstall pytorch torchvision torchaudio pytorch-cuda=11.8 -c pytorch -y
# 然后安装 待实验
# conda install pytorch==2.1.0+cu121 torchvision==2.1.0+cu121 torchaudio==2.1.0+cu121 pytorch-cuda=12.1 -c pytorch -c nvidia

@ -1,5 +0,0 @@
# 首先卸载
conda uninstal torch torchvision torchaudio -y
# 然后安装
conda install pytorch==2.1.0+cu121 torchvision==2.1.0+cu121 torchaudio==2.1.0+cu121 pytorch-cuda=12.1 -c pytorch -c nvidia

@ -151,7 +151,7 @@ def to_data(file_dir, excel):
if __name__ == '__main__':
file_dir = r'./浙江所有地市133行业数据'
p = Pool(4)
p = Pool(6)
for excel in os.listdir((file_dir)):
p.apply_async(func=to_data, args=(file_dir, excel))
p.close()

@ -1,11 +0,0 @@
# 首先卸载
pip3 uninstall torch
pip3 uninstall torchvision
pip3 uninstall torchaudio
# 然后安装
pip3 install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 -f https://download.pytorch.org/whl/cu121/torch_stable.html
# 可临时清华源加速
pip3 install torch==2.1.0+cu121 torchvision==0.16.0+cu121 torchaudio==2.1.0+cu121 -f https://download.pytorch.org/whl/cu121/torch_stable.html -i https://pypi.tuna.tsinghua.edu.cn/simple/
Loading…
Cancel
Save